我想从Python程序中禁用GPU。
我尝试过,但是程序仍在使用GPU。我正在使用Dlib人脸识别。
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
答案 0 :(得分:2)
尝试以下方法:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
文档建议使用无效索引(-1)而不是尝试传递空白值;将此值设置为空白是很多人自己想出的办法,但似乎并不总是奏效。来源:https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars