【那些年踩過的坑】win10 + python 3.6 環境下安裝 pytorch/torchvision
阿新 • • 發佈:2019-01-09
我的安裝環境和版本
- Windows 10
- Python 3.6
- Anaconda 3 64-bit
- Pytorch CPU version
- torchvision
安裝步驟
- 安裝python3.6/3.5(個人用的是3.6。3.5也可以,但我沒試過)
安裝anaconda3 64-bit
注意:- 一定要看好型號哇,windows下pytorch僅支援3.5和3.6,所以只能安裝anaconda3。並且最好安裝64位的。我一開始安裝32位的版本,結果安裝pytorch包時,報錯package not found,可能是沒有32位的包
- 安裝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