1. 程式人生 > >安裝Pytorch的小坑簡記

安裝Pytorch的小坑簡記

嘗試裝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