python中使用指定GPU

Posted sbj123456789

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python中使用指定GPU相关的知识,希望对你有一定的参考价值。

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

 

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

以上是关于python中使用指定GPU的主要内容,如果未能解决你的问题,请参考以下文章

tensorflow中使用指定的GPU及GPU显存

python中使用指定GPU

如何选择指定的GPU来运行CUDA程序?

mmaction2 调用指定GPUmmaction2 使用指定编号的GPU训练模型

pytorch中设定使用指定的GPU

GPU运行Tensorflow的几点建议