2022-凯发ag旗舰厅登录网址下载

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

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旗舰厅登录网址下载为你收集整理的2022-01-14：深度学习中关于显卡的设置的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得凯发ag旗舰厅登录网址下载网站内容还不错，欢迎将凯发ag旗舰厅登录网址下载推荐给好友。