如何禁用Python程序的GPU?

Joh*_*ott 6 python cuda gpu

我想从 Python 程序中禁用 GPU。

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

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
Run Code Online (Sandbox Code Playgroud)

Mic*_*ael 11

试试这个:

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
Run Code Online (Sandbox Code Playgroud)

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

  • 仍然使用GPU。 (3认同)