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
這里我提供一個安裝tensorflow時所用的CUDA對應(yīng)表
安裝cuDNN時版本一定要對應(yīng)正確
安裝完CUDA時要改一下環(huán)境變量
只用添加這個就好，其他三個CUDA會自動為你添加
我的是cuda-11.2
但是官網(wǎng)沒有配套的，直接就挑自己喜歡的下就行Start Locally | PyTorch
這里強(qiáng)調(diào)：千萬不要相信在其他網(wǎng)站上的下載命令。有可能你下載的就是cpu版本
安裝的時候可能會遇到
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)文章
arcgis使用Python腳本進(jìn)行批量截圖功能實(shí)現(xiàn)
最近公司數(shù)據(jù)部那邊有個需求，需要結(jié)合矢量數(shù)據(jù)和影像數(shù)據(jù)，進(jìn)行批量截圖，并且截圖中只能有一個圖斑，還要添加上相應(yīng)的水印，這篇文章主要介紹了arcgis使用Python腳本進(jìn)行批量截圖,需要的朋友可以參考下2023-01-01
python 定時器每天就執(zhí)行一次的實(shí)現(xiàn)代碼
這篇文章主要介紹了python 定時器每天就執(zhí)行一次的實(shí)現(xiàn)代碼，代碼簡單易懂非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了簡單了解python反射機(jī)制的一些知識，反射機(jī)制就是在運(yùn)行時，動態(tài)的確定對象的類型，并可以通過字符串調(diào)用對象屬性、方法、導(dǎo)入模塊，是一種基于字符串的事件驅(qū)動。,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了python中的list字符串元素排序方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-11-11
python并發(fā)2之使用asyncio處理并發(fā)
本篇文章主要介紹了python并發(fā)2之使用asyncio處理并發(fā)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-12-12
在日常編程中,我們經(jīng)常會遇到文件處理的需求,如base64與圖片的相互轉(zhuǎn)換、圖片與像素點(diǎn)的操作、本地文件與二進(jìn)制的互相轉(zhuǎn)換、計算文件的md5以及下載網(wǎng)絡(luò)文件等,這些操作對于處理多媒體數(shù)據(jù)、實(shí)現(xiàn)數(shù)據(jù)的持久化存儲、保證數(shù)據(jù)的完整性和安全性等方面都至關(guān)重要2024-09-09
這篇文章主要介紹了Python編程使用NLTK進(jìn)行自然語言處理詳解，涉及了nltk和開發(fā)環(huán)境的簡單介紹，以及SentencesSegment，SentencesSegment等內(nèi)容，具有一定參考價值，需要的朋友可以了解下。2017-11-11
Diango + uwsgi + nginx項(xiàng)目部署的全過程（可外網(wǎng)訪問）
這篇文章主要給大家介紹了關(guān)于Diango + uwsgi + nginx項(xiàng)目部署的全過程（可外網(wǎng)訪問），文中通過示例代碼將部署的過程介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2018-04-04
這篇文章主要介紹了Python編程pygal繪圖實(shí)例之XY線，具有一定借鑒價值，需要的朋友可以參考下。2017-12-12
Django 連接sql server數(shù)據(jù)庫的方法
這篇文章主要介紹了Django 連接sql server數(shù)據(jù)庫的方法,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-06-06