開發環境:
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
運行結果:(視網速,安裝過程有快有慢,耐心多等待一會)
全程無報錯,到此,安裝就算完成了。
最後,驗證:
導入無報錯,安裝成功。