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

原文链接

1 间接在终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

2 python 代码中设定：

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

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

代码的路