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