【pytorch】如何使用指定GPU

pytorch 中使用指定GPU的方法有很多，这里只列出两种最常用的方法，一般就足够使用了

方法一

python代码中指定

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

方法二

在终端中设定该程序可见的GPU

CUDA_VISIBLE_DEVICES=1,2,3 python my_script.py

参考资料：

PyTorch指定GPU的方法 - 简书

（原）PyTorch中使用指定的GPU - darkknightzh - 博客园