【问题标题】:How to disable GPU for Python program?如何为 Python 程序禁用 GPU?
【发布时间】: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 cuda gpu


    【解决方案1】:

    试试这个:

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

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

    【讨论】:

    • 如果您从命令行而不是通过 os.environ 显式设置 CUDA_VISIBLE_DEVICES 会发生什么?
    • 仍在使用 GPU。
    • 一定有办法完全禁用/隐藏/移除GPU?
    • 对我来说,在实际代码执行之前使用export CUDA_VISIBLE_DEVICES="" 似乎几乎每次都有效
    猜你喜欢
    • 2021-04-27
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-04-30
    • 2020-10-10
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多