安裝Pytorch的小坑簡記
阿新 • • 發佈:2019-01-27
嘗試裝pytorch,官網上的介紹比較簡單,但是安裝過程中出現了一些問題,記錄一下,不一定對所有機器都適用。
寫在前面
環境如下:Ubuntu16.04(自帶python2.7)
一、版本問題
最開始使用是Anaconda3-4.3.1,其自帶的是python3.6,Pytorch官網上雖然給出了python3.6對應的命令,但是在本機安裝不成功,最後安裝成功的版本如下:
Anaconda3-4.2.0,自帶的是python3.5,相應的pytorch對應的python版本選擇python3.5的。
二、使用pip3 install torchvision出現錯誤
1、問題描述
在使用pip3 install http://download.pytorch.org/whl/cpu/torch-0.3.1-cp35-cp35m-linux_x86_64.whl
成功安裝後,使用pip3 install torchvision
的時候提示如下錯誤資訊:
Collecting torchvision
Using cached torchvision-0.2.0-py2.py3-none-any.whl
Collecting numpy (from torchvision)
Using cached numpy-1.14.2-cp35-cp35m-manylinux1_x86_64.whl
Collecting torch (from torchvision)
Using cached torch-0.1.2.post1.tar.gz
Complete output from command python setup.py egg_info:
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "/tmp/pip-build-kqdxllrv/torch/setup.py", line 11, in <module>
raise RuntimeError(README)
RuntimeError: PyTorch does not currently provide packages for PyPI (see status at https://github.com/pytorch/pytorch/issues/566).
Please follow the instructions at http://pytorch.org/ to install with miniconda instead.
----------------------------------------
Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-kqdxllrv/torch/
2、解決方案
將pip3 install torchvision
命令換成一下命令即可:
pip3 install --no-deps torchvision