Windows10安裝Pytorch步驟記錄

ygpGoogle發表於2020-11-27

安裝Anaconda並且建立Pytorch環境

下載Anaconda

先從anaconda官網,選擇匹配自己系統的版本,安裝anaconda;
注意!如果你係統已經裝了python的話,不要勾選add anaconda to the system variable選項,這個的意思是自動幫你加到環境變數,但是我們自己加會比較好,要不然有可能會和原有的python衝突
同時可以自行選擇是否安裝vscode,如果之前已經安裝了那就不用,如果沒安裝但不想用vscode那也別勾選。

配置環境變數

將anaconda的安裝目錄和其下的scripts目錄新增到系統的環境變數中,紅圈裡是我自己的安裝路徑,大家需要根據自己安裝的情況做更改。
注意是新建
在這裡插入圖片描述

anaconda的命令列視窗

在這裡插入圖片描述

配置新增新的下載源

這是因為很多conda依賴包的下載連結是外網,下載很慢,還經常斷,用清華映象會快很多

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

這樣就把清華映象站中anaconda各種依賴包的映象原新增進了conda命令中
在這裡插入圖片描述

新增啟用新的python環境

還是開啟anaconda的prompt命令列視窗,鍵入

conda create -n pytorch python=3.6

其中pytorch是自行設定的環境名,隨便取名就好,python=X.X是在這個環境中安裝X.X版本的python,這裡用3.6
隨後會提示是否安裝,鍵入y即可

安裝結束後在anaconda prompt中鍵入

activate pytorch

這樣就啟用了這個新環境,此時應該可以看到:
在這裡插入圖片描述

需要注意的是,該新環境中只有很少的python依賴庫,剩下的要看自己需要利用pip install或conda install安裝,由於這裡我們是配置pytoch的環境,因此沒有什麼額外需要的,就不管它,接下來介紹一些對環境的操作:

啟用環境已經介紹過了
安裝包:conda install -n your_env_name xxx;
或者啟用環境後利用pip install xxx
退出環境:deactivate your_env_name
刪除環境:conda remove -n your_env_name --all
檢視所有環境:conda env list

至此,pytorch環境建立成功

利用conda安裝pytorch

新增pytorch映象源

將清華映象源新增到conda install的搜尋channels中
開啟anaconda prompt命令列視窗,鍵入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

正式安裝pytorch

進入pytorch官網,選擇適合自己電腦系統的版本,在這裡檢視不同顯示卡驅動所對應的cuda版本,我電腦沒有GPU,就只能選NONE了
在這裡插入圖片描述

conda install pytorch torchvision torchaudio cpuonly

收集好下載包,按y進入安裝即可。

然而,我就是這裡遇到了坑,我的顯示卡驅動版本對應9.2的cuda,但是報錯Solving environment: failed with initial frozen solve. Retrying with flexible solve,之後顯示package not found error,如果各位夥伴也出現相同問題,那別再糾結conda安裝了,不管怎樣離線安裝也好、搜尋包也好,都是不行的,但是用pip可以順利解決;當然,如果沒報錯順利安裝那是最好。

檢視是否安裝成功

在pytoch環境下在prompt中鍵入

import torch
import torchvision
print(torch)
#結果為<module 'torch' from 'xxx\\torch\\__init__.py'>
#xxx為安裝的路徑

如果沒有報錯,則代表安裝成功,開始進擊吧!但推薦看一下下一個模組,有一個很方便的pip映象源

利用pip安裝

將清華映象源新增到pip的搜尋目錄中,開啟anaconda prompt,鍵入

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

這是清華映象中python依賴庫的部分,基本涵蓋了所有庫,而且下載超快,鍵入這行命令後就會從中下載之後pip install的庫

進入pytorch官網,選擇適合自己電腦系統的版本,在這裡檢視不同顯示卡驅動所對應的cuda版本
在這裡插入圖片描述

開啟anaconda prompt,啟用pytorch環境

activate pytorch

再鍵入該command,同時在command後面加上-i https://pypi.tuna.tsinghua.edu.cn/simple,即鍵入

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

按y確認安裝即可

不出所料,我又在這裡遇到了坑,不管我怎樣配置,依然在從官網下載安裝包,並且下載速度僅有幾十kb/s,遇到這種情況的夥伴別急,繼續往下看,很輕鬆就可以解決,沒有遇到的真的羨慕你萌(哭)
解決方法:
1.找到這個command裡涉及的官網下載連結
2.迅雷下載!
3.為了一致性,我將其下載到了pytorch環境下的site-packages中,我的路徑是C:\apps\Anaconda\envs\pytorch\Lib\site-packages
4.開啟anaconda prompt
5.啟用pytoch環境
6.鍵入

pip install C:\apps\Anaconda\envs\pytorch\Lib\site-packages\torch-1.4.0+cu92-cp37-cp37m-win_amd64.whl

其中路徑已經說過,最後的包名稱是你下載的那個,注意要加.whl
7。鍵入y確認安裝
8.再次鍵入

pip install torchvision	

這是安裝torchvision及pytoch的各種依賴庫,否則import torch依然會報錯

驗證是否安裝成功:
在pytoch環境下在prompt中鍵入

import torch
import torchvision
print(torch)
#結果為<module 'torch' from 'xxx\\torch\\__init__.py'>
#xxx為安裝的路徑

至此,安裝完成,而且以後的pip install都會很快~
衝起來吧!!!

相關文章