三步安裝pytorch-cpu(win10下安裝pytorch-cpu)

開發環境:

win10 x64位

anaconda3-4.0.0

虛擬環境python版本 3.6.5(安裝過程中根據你安裝的pytorch版本可能需要升級)

 

前提:anaconda已配置好環境變量,使用conda切換到需要安裝pytorch的虛擬環境。

 

第一步:anaconda改爲國內源(如果已經改過,請檢查添加的國內源中是否有pytorch的源(下方代碼中倒數第二行),如果沒有請添加國內pytorch源)

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/conda-forge/ 
 
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
 
conda config --set show_channel_urls yes

第二步:搜索可用版本

conda search pytorch-cpu

執行結果:(因爲時間不一樣可能會有新的版本發佈,所以搜索結果可能也不一樣)

從搜索結果可以看到,共搜索到0.4.0、0.4.1、1.0.0、1.1.0四個版本,但並不是版本越高越好,版本越高可能穩定性越差(猜測),所以儘量找中間版本或者根據需要進行安裝,這裏我選擇安裝0.4.1版本。

第三步:安裝

# 先安裝pytorch-cpu
conda install pytorch-cpu=0.4.1

# 然後安裝依賴torchvision-cpu,這裏不需要指定版本
conda install torchvision-cpu

運行結果:(視網速,安裝過程有快有慢,耐心多等待一會)

 

 

全程無報錯,到此,安裝就算完成了。

最後,驗證:

 

導入無報錯,安裝成功。

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章