如何为GPU程序禁用GPU?

时间:2019-09-26 13:14:19

标签: python cuda gpu

我想从Python程序中禁用GPU。

我尝试过,但是程序仍在使用GPU。我正在使用Dlib人脸识别。

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

1 个答案:

答案 0 :(得分:2)

尝试以下方法:

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

文档建议使用无效索引(-1)而不是尝试传递空白值;将此值设置为空白是很多人自己想出的办法,但似乎并不总是奏效。来源:https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars