1. 程式人生 > 其它 >安裝anaconda 並配置pytorch的環境

安裝anaconda 並配置pytorch的環境

1.安裝pytorch如果不需要gpu上執行的話,直接安裝anaconda(不用安裝cuda,cuDNN)
Anaconda 下載地址:https://www.anaconda.com/products/distribution
直接可以下載 64 位,python 3.9版本的:


(1)、安裝步驟:



(2、)配置環境變數:

(3)測試是否安成功:
win+R, cmd

先不著急建立虛擬環境,從開始搜尋Anaconda Navigator ,點選開啟Anaconda Navigator,啟動一下jupyter Notebook 看能不能進入網頁

(4)、在anaconda中建立虛擬環境
在開始搜尋anaconda prompt

conda create -n my_pytorch python=3.9 -n後面是虛擬環境的名字(自己取)

檢視建立虛擬環境:conda info -e

在下載pytorch前可以先下載numpy(大部分不需要,下載pytorch會自帶numpy(自己下的時候))
先啟動虛擬環境,conda activate my_pytroch回車,前面不在顯示base會顯示你自己建立的虛擬環境的名字。
然後安裝numpy,輸入conda install numpy (這個下載不需要多久) 然後下載對應的pytorch
官網:https://pytorch.org/get-started/locally/


下載之前 先可以檢視一下自己的cuda版本 , 指令nvidia-smi(anaconda和命令提示符都可以)(我的是11.4版本,裝的是11.3)

也可以下載cpu版的(賊慢,核心經常掛掉,一個多項式函式的擬合都會崩潰),沒有獨顯也可以裝gpu(電腦自帶核顯,但沒啥用,也慢得很)
其他版本的pytorch(從裡面去找) :https://pytorch.org/get-started/previous-versions/
接下來就是下載,直接下載的話很慢,新增清華映象源地址:(先退出自己的虛擬環境, 指令 conda deactivate 然後新增下面三條指令 )
conda config --add channels

http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
檢視這些channel : conda config --set show_channel_urls yes

再進入自己的虛擬的環境 conda activate my_pytorch 複製官網的 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

這個下載真的很隨緣,反正自己當時下載的時候這些源都沒有用,下載很慢,有些包下載完成了,還有一些沒有下載完成,尤其pytorch包太大。
可以直接下載pytorch的離線包(當然下載成功就不用了)。
從清華映象源的地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ 下載離線包,我下載是11.3

然後從anaconda進入虛擬環境 conda install D:\Software\pytorch-1.11.0-py3.9_cuda11.3_cudnn8_0.tar.bz2 (install後 + 自己包下載的位置) 當時也不記得報錯沒有,有些沒有下載的包可以不用理會,需要的時候可以在下載。

驗證pytorch是否安裝成功,進入自己的虛擬環境,conda activate my_pytorch, (打字真累,上圖吧)

沒有報錯說明你的pytorch安裝是成功的。報錯的話,你得自己檢視看原因,應該是安裝pytorch那裡出了問題
下面就是要安裝cuda和cuDNN,這篇部落格有點長,我得重新再寫一篇安裝cuda和cuDNN

pytorch 之安裝cuda和cuDNN
https://www.cnblogs.com/panxubin1011703/p/16150548.html