pytorch-gpu安裝的經(jīng)驗(yàn)與教訓(xùn)
首先說明
本文并不是安裝教程，網(wǎng)上有很多，這里只是自己遇到的一些問題
我是以前安裝的tensorflow-gpu的，但是發(fā)現(xiàn)現(xiàn)在的學(xué)術(shù)論文大部分都是用pytorch復(fù)現(xiàn)的，因此才去安裝的pytorch-gpu
查看自己安裝的CUDA
nvcc -V
這里我提供一個(gè)安裝tensorflow時(shí)所用的CUDA對應(yīng)表
安裝cuDNN時(shí)版本一定要對應(yīng)正確
安裝完CUDA時(shí)要改一下環(huán)境變量
只用添加這個(gè)就好，其他三個(gè)CUDA會自動(dòng)為你添加
我的是cuda-11.2
但是官網(wǎng)沒有配套的，直接就挑自己喜歡的下就行Start Locally | PyTorch
這里強(qiáng)調(diào)：千萬不要相信在其他網(wǎng)站上的下載命令。有可能你下載的就是cpu版本
安裝的時(shí)候可能會遇到
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
不要擔(dān)心，不用管它，耐心等一等，就會出現(xiàn)（y/n）選擇y就可以了
這里要看一眼你的pytorch是不是cuda版本的，不是要重新下載
測試安裝是否成功
import torch print(torch.__version__) #注意是雙下劃線 print(torch.version.cuda) print(torch.cuda.is_available()) print(torch.cuda.get_device_name())
當(dāng)然你這些都做完了，還是出現(xiàn)了False
失敗后的操作
打開你的anaconda
將這里關(guān)于pytorch的信息全部刪除，再按照前面的流程走一遍，相信你自己一定會出現(xiàn)
到此這篇關(guān)于pytorch-gpu安裝的經(jīng)驗(yàn)與教訓(xùn) 的文章就介紹到這了,更多相關(guān)pytorch-gpu安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python正則表達(dá)式實(shí)現(xiàn)截取成對括號的方法
這篇文章主要介紹了Python正則表達(dá)式實(shí)現(xiàn)截取成對括號的方法,涉及Python正則匹配相關(guān)操作技巧,需要的朋友可以參考下2017-01-01
python 五子棋如何獲得鼠標(biāo)點(diǎn)擊坐標(biāo)
這篇文章主要介紹了python 五子棋如何獲得鼠標(biāo)點(diǎn)擊坐標(biāo),文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11
今天小編就為大家分享一篇OpenCV2從攝像頭獲取幀并寫入視頻文件的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-08-08
使用Python實(shí)現(xiàn)XLS和XLSX之間的相互轉(zhuǎn)換
在日常工作中,我們經(jīng)常需要處理和轉(zhuǎn)換不同格式的Excel文件,以適應(yīng)不同的需求和軟件兼容性,Excel文件的兩種常見格式是XLS（Excel 97-2003）和XLSX（Excel 2007及以上版本）,本文將詳細(xì)介紹如何使用Python在XLS和XLSX格式之間進(jìn)行轉(zhuǎn)換,需要的朋友可以參考下2024-09-09
pytorch DataLoader的num_workers參數(shù)與設(shè)置大小詳解
這篇文章主要介紹了pytorch DataLoader的num_workers參數(shù)與設(shè)置大小詳解，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2021-05-05
Python機(jī)器學(xué)習(xí)之底層實(shí)現(xiàn)KNN
今天給大家?guī)淼氖顷P(guān)于Python機(jī)器學(xué)習(xí)的相關(guān)知識,文章圍繞著Python底層實(shí)現(xiàn)KNN展開,文中有非常詳細(xì)的解釋及代碼示例,需要的朋友可以參考下2021-06-06
在Python中,我們可以通過__set_name__方法來實(shí)現(xiàn)一些特殊的操作,本文主要介紹如何在Python中實(shí)現(xiàn)__set_name__方法,并且給出一些實(shí)際應(yīng)用的示例,感興趣的可以了解一下2024-01-01
Windows下Python的Django框架環(huán)境部署及應(yīng)用編寫入門
這篇文章主要介紹了Windows下Python的Django框架環(huán)境部署及程序編寫入門,Django在Python的框架中算是一個(gè)重量級的MVC框架,本文將從程序部署開始講到hellow world web應(yīng)用的編寫,需要的朋友可以參考下2016-03-03
Python中實(shí)現(xiàn)列表的逆序、復(fù)制與清除的幾種常見方法
本文介紹了Python中列表的逆序、復(fù)制和清除操作,通過reverse()方法、切片、copy()方法和clear()方法,我們可以輕松地對列表進(jìn)行這些操作2024-12-12
關(guān)于Python中的向量相加和numpy中的向量相加效率對比
今天小編就為大家分享一篇關(guān)于Python中的向量相加和numpy中的向量相加效率對比，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08