Ubuntu安装torchvision

pytorch是torch移植到python上的版本&＃xff0c;作为python3的第三方库&＃xff0c;我们首先想到的就是利用pip来进行安装&＃xff0c;但是直接使用

pip3 install pytorch  &＃xff08;因为我使用的是python3&＃xff0c;因此此处为pip3&＃xff09;&＃xff0c;会报如下错误&＃xff1a;

提示我们要去pytorch的官网进行下载&＃xff0c;打开官网&＃xff0c;可以按照安装方式&＃xff08;pip ,conda,source&＃xff09;和python版本&＃xff0c;cuda版本来进行安装&＃xff0c;由于本国特殊的网络国情&＃xff0c;一般是打不开&＃xff0c;或者是无法查看 run the command 之后的指令的。

在题主进行科学上网以后&＃xff0c;得到了利用pip进行安装的相关的指令&＃xff0c;&＃xff08;该指令为linux,pip,python3.6,none-cuda&＃xff09;

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl &＃xff08;linux下&＃xff0c;即使cuda选择none&＃xff0c;还是默认安装cuda75版本&＃xff09;

pip install torchvision

却发现torchvision已经不再支持pip&＃xff0c;pytorch可以正常使用

看提示&＃xff0c;建议我们选择miniconda进行安装&＃xff0c;于是乎转战conda&＃xff0c;由于我们学校提供了官方的镜像库&＃xff0c;所以我选择直接安装了anaconda&＃xff0c;如何在ubuntu上安装anaconda有很多文章可以参考&＃xff0c;可以之前前往  https://www.continuum.io/downloads/  进行安装。

利用conda进行安装的指令如下&＃xff1a;

cuda7.5 

conda install pytorch torchvision -c soumith

cuda8.0

conda install pytorch torchvision cuda80 -c soumith

然而&＃xff01;利用conda进行安装时会报condaHttpError错误&＃xff0c;起初我认为是我国特殊的网络环境造成的&＃xff0c;因此&＃xff0c;将conda的源切换为我们学校的镜像源&＃xff0c;仍然会报HttoError,但是安装别的第三方库可以成功。此时我是绝望的。这2个第三方包管理器太不靠谱&＃xff0c;我在starkoverflow和github上找了一下&＃xff0c;有很多人碰到了类似的问题&＃xff0c;也有一些回答&＃xff0c;但经过我的尝试&＃xff0c;没有一个能解决我的问题。

最终解决方法----源码编译

源码包在github上的托管地址为

https://github.com/pytorch/vision

https://github.com/pytorch/pytorch

分别克隆2个包的源代码进行安装&＃xff1a;

git clone git&＃64;github.com:pytorch/pytorch.git

git clone git&＃64;github.com:pytorch/vision.git

然后&＃xff0c;cd到克隆的目录下

cd pytorch

python3 setup.py install

cd vision

python3 setup.py install

注意&＃xff0c;若安装时提示找不到cmake&＃xff0c;直接sudo apt install cmake 即可

安装完成之后&＃xff0c;进入ipython3&＃xff0c;查看是否安装成功&＃xff08;直接在python命令行查看也可&＃xff09;

之后就可以畅游pytorch和pytorchvision了。