如何判斷pytorch是否支持GPU加速
更新時間：2021年06月01日 09:40:34 作者：抬頭仰望-y
這篇文章主要介紹了判斷pytorch是否支持GPU加速的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
加上這句代碼：
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()
返回當前設(shè)備索引；
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
pygame.time 時間控制模塊，是 Pygame 中使用頻率較高的模塊，其主要功能是管理時間和游戲幀數(shù)率，本文就來詳細的介紹一下，感興趣的可以了解一下2021-11-11
接上一文用iptables讓SSH服務(wù)對陌生人說不。還是有點擔心這個學(xué)期內(nèi)，nginx可能會因為系統(tǒng)各種原因而出現(xiàn)異常退出，導(dǎo)致Web服務(wù)暫停。所以，又來了一個方案2012-05-05
pytest+request框架寫接口測試自動化,使用yaml文件配置更方便管理用例中的數(shù)據(jù),本文主要介紹了pytest+request框架中yaml配置文件使用,具有一定的參考價值,感興趣的可以了解一下2024-01-01
本篇文章主要介紹了詳解用TensorFlow實現(xiàn)邏輯回歸算法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-05-05
Python中列表元素轉(zhuǎn)為數(shù)字的方法分析
這篇文章主要介紹了Python中列表元素轉(zhuǎn)為數(shù)字的方法,結(jié)合實例形式對比分析了Python列表操作及數(shù)學(xué)運算的相關(guān)技巧,需要的朋友可以參考下2016-06-06
今天小編就為大家分享一篇python實現(xiàn)簡單的單變量線性回歸方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
今天小編就為大家分享一篇Python全面分析系統(tǒng)的時域特性和頻率域特性，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
學(xué)python的第二天，想寫個東西出來玩玩，于是就寫了這個，供那些才學(xué)一天的參考參考也行2013-08-08
在本篇文章里小編給大家整理的是一篇關(guān)于Django項目在pycharm新建的步驟方法，有興趣的朋友們可以學(xué)習(xí)參考下。2021-03-03
Python GUI Tkinter簡單實現(xiàn)個性簽名設(shè)計
這篇文章主要為大家詳細介紹了Python GUI Tkinter簡單實現(xiàn)個性簽名設(shè)計，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-06-06