pytorch指定使用的单个GPU

Posted Shaw_喆宇

tags:

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

1.pycharm里直接在代码中加入下面

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

2.在终端指定使用

CUDA_VISIBLE_DEVICES=2 python xxx.py

内容源自:https://www.cnblogs.com/darkknightzh/p/6836568.html

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