1. 程式人生 > >【那些年踩過的坑】win10 + python 3.6 環境下安裝 pytorch/torchvision

【那些年踩過的坑】win10 + python 3.6 環境下安裝 pytorch/torchvision

我的安裝環境和版本

  • Windows 10
  • Python 3.6
  • Anaconda 3 64-bit
  • Pytorch CPU version
  • torchvision

安裝步驟

  • 安裝python3.6/3.5(個人用的是3.6。3.5也可以,但我沒試過)
  • 安裝anaconda3 64-bit
    注意:

    1. 一定要看好型號哇,windows下pytorch僅支援3.5和3.6,所以只能安裝anaconda3。並且最好安裝64位的。我一開始安裝32位的版本,結果安裝pytorch包時,報錯package not found,可能是沒有32位的包
    2. 安裝anaconda時推薦選“only install for one user”,最好別選“install for all users”。很多選擇了multi-user的使用者都出現了安裝/更新包時報錯permission denied.
  • 成功執行完第一行命令後,痛苦又開始了!
    我用pip3安裝torchvision竟然報錯!
    報錯
    報錯裡建議我用conda安裝,然後嘗試了各種“conda install xxx torchvision xxx”,均以失敗告終。我甚至嘗試了下載whl檔案直接安裝,都沒成功。
    最後,我竟然鬼使神差地用了pip而不是pip3,安裝成功了!如果你同時安裝了python2.x,不建議嘗試pip安裝,容易搞混。

pip install torchvision

安裝成功

  • 最後測試一下是否安裝成功
    天真如我竟然輸入了 “import pytorch”
    下面才是正確的程式碼
import
torch import torchvision