1 直接在终端中设定：
CUDA_VISIBLE_DEVICES=1 python main.py
2 python 代码中设定：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
学习更多编程知识，请关注我的公众号：
1 直接在终端中设定：
CUDA_VISIBLE_DEVICES=1 python main.py
2 python 代码中设定：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
学习更多编程知识，请关注我的公众号：
本文标题：PyTorch指定GPU进行训练
本文链接：https://www.haomeiwen.com/subject/vbvegrtx.html
网友评论