【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 - 博客园

Guess you like

Origin blog.csdn.net/flyconley/article/details/119134563