PyTorch指定GPU进行训练

【PyTorch指定GPU进行训练】原文链接
1直接终端中设定:

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定:
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"



学习更多编程知识,请关注我的公众号:
代码的路
PyTorch指定GPU进行训练
文章图片

    推荐阅读