如何禁用Python程序的GPU?

2024-04-25 04:32:03 发布

您现在位置:Python中文网/ 问答频道 /正文

我想从Python程序中禁用GPU。你知道吗

我试过了,但程序仍然使用GPU。我正在使用Dlib人脸识别。你知道吗

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""

Tags: 程序idgpuosdeviceenvironorderissue
1条回答
网友
1楼 · 发布于 2024-04-25 04:32:03

请尝试以下操作:

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

文档建议使用无效索引(-1),而不是试图传递一个空值;将这个值设置为空值是许多人自己想到的,但似乎并不总是奏效。资料来源:https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars

相关问题 更多 >