pytorch 中使用指定GPU的方法有很多,这里只列出两种最常用的方法,一般就足够使用了
方法一
python代码中指定
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
方法二
在终端中设定该程序可见的GPU
CUDA_VISIBLE_DEVICES=1,2,3 python my_script.py
参考资料:
pytorch 中使用指定GPU的方法有很多,这里只列出两种最常用的方法,一般就足够使用了
python代码中指定
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
在终端中设定该程序可见的GPU
CUDA_VISIBLE_DEVICES=1,2,3 python my_script.py
参考资料: