西寧做網(wǎng)站的公司力請(qǐng)君博d百度網(wǎng)站關(guān)鍵詞排名查詢
下載安裝Anaconda
首先需要下載Anaconda,可以到官網(wǎng)Anaconda官網(wǎng)或者這里提供一個(gè)鏡像網(wǎng)站去下載anaconda鏡像網(wǎng)站
安裝步驟可參考該文章:Anaconda安裝步驟,本篇不再贅述
注意環(huán)境變量的配置,安裝好Anaconda之后一定要在環(huán)境變量中添加以下內(nèi)容:
創(chuàng)建虛擬環(huán)境
安裝好Anaconda后,按下Windows按鈕,找到Anaconda Prompt
,點(diǎn)擊打開
打開后的界面如下:
接下來介紹幾個(gè)conda命令
- 查看當(dāng)前存在哪些虛擬環(huán)境:
conda env list
conda info -e
以上是我建立的虛擬環(huán)境,如果是第一次下載,則只會(huì)有base這一個(gè)虛擬環(huán)境
- 新建一個(gè)虛擬環(huán)境
conda create -n pytorch39 python=3.9
其中pytorch39是你的虛擬環(huán)境的名字,你也可以取其他方便記住的名字,比如python39、pytorch等,最后的python=3.9是指定python的版本,我用的是3.9
- 創(chuàng)建完成之后需要激活該環(huán)境
activate pytorch39
輸入以上命令可以激活pytorch虛擬環(huán)境,激活之后,簽名的base換成pytorch39,就表示已經(jīng)進(jìn)入到該虛擬環(huán)境了
到此虛擬環(huán)境安裝完畢!
注意:有時(shí)可能會(huì)出現(xiàn)一個(gè)問題,就是你明明已經(jīng)按照這個(gè)方法創(chuàng)建了虛擬環(huán)境,但是在Anaconda的envs目錄中依然沒有所創(chuàng)建的虛擬環(huán)境,這時(shí),在虛擬環(huán)境下輸入 conda info -e
查看虛擬環(huán)境,以及它的位置,正常的應(yīng)該如下圖:
但是有可能你所創(chuàng)建的虛擬環(huán)境的位置是C:\Users\Lenovo.conda\envs,在D:\Anaconda\envs(你安裝Anaconda的目錄)里面找不到,這個(gè)時(shí)候,到C:\Users\Lenovo目錄下找到.condarc文件,打開后在文件里添加以下命令:
envs_dirs:- D://Anaconda//envs
保存之后再次創(chuàng)建虛擬環(huán)境,就會(huì)存儲(chǔ)到Anaconda的envs目錄下,方便查找和管理。
下載GPU的pytorch
首先在Anaconda Prompt中輸入nvidia-smi
查看你的CUDA的版本,觀察Driver Version的值是否大于400,如果小于請(qǐng)更新顯卡驅(qū)動(dòng)。
進(jìn)入到pytorch官網(wǎng):pytorch官網(wǎng),往下翻找到install previous version of PyTorch,可以查看之前的版本。
找到符合你cuda版本的命令,首先使用conda命令進(jìn)行安裝,比如我的cuda版本是10.2,往下翻可以找到對(duì)應(yīng)的命令
在pytorch39虛擬環(huán)境中輸入該命令就可以進(jìn)行下載
如果下載不成功,多嘗試幾種方法,比如:
- 不指定版本,conda會(huì)自動(dòng)安裝較新版本,可以輸入
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
- 如果下載速度較慢,可以去掉后面的
-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2
下載完成后在pytorch39虛擬環(huán)境下輸入以下python進(jìn)入到python中,再輸入以下命令查看是否下載成功:
import torch
torch.cuda.is_available()
如果返回True,則安裝成功
但是!!!
很多情況下會(huì)返回False,是因?yàn)閏onda自動(dòng)安裝的話可能會(huì)安裝CPU版本的,而不是GPU版本
如果之前配置過清華源使用pytorch官網(wǎng)的命令會(huì)報(bào)錯(cuò),清華源是國內(nèi)的鏡像,pytorch是國外的,因此連接會(huì)出現(xiàn)問題,可以將之前的.condarc文件夾移除掉載嘗試
可以再嘗試在后面添加-c nvidia
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch -c nvidia
上一步的詳細(xì)內(nèi)容在這里
如果還是CPU的版本,再進(jìn)行以下方法,親測(cè)有效!!
我的cuda是10.2版本,所以這里只演示10.2版本的pytorch的安裝,其他版本的方法也類似
因?yàn)閏onda會(huì)自動(dòng)下載CPU的,而要做深度學(xué)習(xí)的實(shí)驗(yàn)需要用到GPU的,所以你可以去自己下載pytorch的whl文件下載以下三個(gè)文件
- 下載torch:
- 下載torchaudio
- 下載torchvision
將下載后的這三個(gè)文件放到D:盤的根目錄下
在pytorch39虛擬環(huán)境中依次輸入以下幾行代碼,就可以將這三個(gè)whl文件安裝到虛擬環(huán)境中
pip install D:\torch-2.1.1+cu121-cp39-cp39-win_amd64.whl
pip install D:\torchvision-0.16.1+cu121-cp39-cp39-win_amd64.whl
pip install D:\torchaudio-2.1.1+cu121-cp39-cp39-win_amd64.whl
安裝后再次檢驗(yàn)
python
import torch
print(torch.cuda.is_available()) #輸出True表示可用GPU版本torch
print(torch.__version__)#輸出對(duì)應(yīng)torch版本號(hào)
該方法詳情內(nèi)容在這里
最后不出意外返回的應(yīng)該就是True了,恭喜你安裝成功了!