乐趣区

关于程序员:PyTorch指定GPU进行训练

原文链接

1 间接在终端中设定:

CUDA_VISIBLE_DEVICES=1 python main.py

2 python 代码中设定:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

 
 

学习更多编程常识,请关注我的公众号:

代码的路

退出移动版