# 1：
torch.cuda.set_device(1)
# 2：
device = torch.device("cuda:1")
# 3：（官方推荐）
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
（同时调用两块GPU的话）
os.environ["CUDA_VISIBLE_DEVICES"] = '1,2'
pytorch 指定GPU训练
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。