加上這句代碼：

print torch.cuda.is_available()

判斷完畢！說說在pytorch中如何查看gpu信息吧~

為什么將數(shù)據(jù)轉(zhuǎn)移至GPU的方法叫做.cuda而不是.gpu，就像將數(shù)據(jù)轉(zhuǎn)移至CPU調(diào)用的方法是.cpu？這是因為GPU的編程接口采用CUDA，而目前并不是所有的GPU都支持CUDA，只有部分Nvidia的GPU才支持。

PyTorch未來可能會支持AMD的GPU，而AMD GPU的編程接口采用OpenCL，因此PyTorch還預(yù)留著.cl方法，用于以后支持AMD等的GPU。

torch.cuda.is_available()

cuda是否可用；

torch.cuda.device_count()

返回gpu數(shù)量；

torch.cuda.get_device_name(0)

返回gpu名字，設(shè)備索引默認從0開始；

torch.cuda.current_device()

返回當(dāng)前設(shè)備索引；

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持本站。

美國服務(wù)器租用

版權(quán)聲明：本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場，如有內(nèi)容涉嫌侵權(quán)，請聯(lián)系alex-e#qq.com處理。