【发布时间】:2020-01-26 18:45:26
【问题描述】:
我想从 Python 程序中禁用 GPU。
我试过了,但程序仍在使用 GPU。我正在使用 Dlib 人脸识别。
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
【问题讨论】:
我想从 Python 程序中禁用 GPU。
我试过了,但程序仍在使用 GPU。我正在使用 Dlib 人脸识别。
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
【问题讨论】:
试试这个:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
文档建议使用无效索引 (-1) 而不是尝试传递空白值;将此值设置为空白是各种人自己提出的,但似乎并不总是有效。来源:https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars
【讨论】:
export CUDA_VISIBLE_DEVICES="" 似乎几乎每次都有效