2022-ag凯发k8国际

如何将其他显卡设置为主卡呢？

os.environ["cuda_visible_devices"] = "2,0,1"#此时默认主卡为2号实际显卡 model = torch.nn.dataparallel(model, device_ids=[0,1]).cuda()#此时只用上0和1号运算显卡，实际上是2号和0号实际显卡

torch.cuda主要函数

import torch print(torch.cuda.is_available())#判断gpu是否可用 os.environ["cuda_visible_devices"] = '1,2'#进行指定使用设备 print(torch.cuda.device_count())#输出有几个可用显卡，应为2 print(torch.cuda.current_device())#0，pytorch感知的编号还是从device:0开始

参考文章

总结

以上是ag凯发k8国际为你收集整理的2022-01-14：深度学习中关于显卡的设置的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得ag凯发k8国际网站内容还不错，欢迎将ag凯发k8国际推荐给好友。