如何將其他顯卡設置為主卡呢？

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開始

參考文章

總結

以上是生活随笔為你收集整理的2022-01-14：深度学习中关于显卡的设置的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。