PyTorch指定GPU进行训练

原文链接

1 直接在终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定：

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

   

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

代码的路