概述
pytorch上配置使用双显卡
首先那你要有两块或多块显卡，其次torch.cuda.is_available()要返回为true
// 可见设置，环境变量使得指定设备对CUDA应用可见
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
//选择第一块显卡开始载入数据
device = torch.device("cuda: 0")
model.to(device)
//显卡大于1块时，device_ids选择模型载入数据对应的显卡
if torch.cuda.device_count() > 1:
model= nn.DataParallel(model,device_ids=[0,1])
最后
以上就是知性板栗为你收集整理的pytorch上配置使用双显卡或多显卡的全部内容，希望文章能够帮你解决pytorch上配置使用双显卡或多显卡所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复