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