时间戳

首页 > 解决方案 > 如何为 Python 程序禁用 GPU？

python - 如何为 Python 程序禁用 GPU？

问题描述

我想从 Python 程序中禁用 GPU。

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

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

标签： pythoncudagpu

解决方案

试试这个：

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

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

推荐阅读