ubuntu16.04 gtx1080ti 安装pytorch

         最近想用pythorch训练yolo，所以搭建了一下pytorch的开发环境

        pytorch安装虽然简单但是也是有很多坑的！首先我选择的是cuda9.0版本的，我的显卡驱动是nvida-384.130.安装方法在我上一篇博客中有介绍。本来想装cuda8.0版本的，但是因为pytorch的cuda8.0没有torchvision所以选择了cuda9.0

        首先安装pytorch有很多方法。可以通过conda或者pip安装

conda安装虽然简单方便根据 官网 安装教程的一条命令安装，官网上也介绍了其他的安装方式，我也是参考着官方提供的方法进行安装的：

            

    conda install pytorch=0.4.1 cuda90 -c pytorch

这个访问的是国外的镜像源，基本是安装失败的。网上会有很多更换清华源的方法，但即使是添加了国内的源，也不一定能安装成功，首先要确定所添加的源里面是否包含你所需的包。所以这里直接跳过这种方法。

通过pip安装。

从网站上下载所需的包。如图所示：

 执行：

pip3 install torch-1.1.0-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision-0.3.0-cp35-cp35m-manylinux1_x86_64.whl

 安装就完成了。

测试：当出现true的时候就pytorch cuda版本就安装成功了。

~/pytorch/cuda90$ python3.5
Python 3.5.2 (default, Jan 26 2021, 13:30:48)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.cuda.is_available()
True

总结：

        1.conda安装由于源的问题安装很慢，稍大一点的安装包几乎就是安装失败的。

        2.更换国内镜像源要保证所需要的的安装包是有的，不然也是安装失败的

        3.pip安装的时候要查看你的python版本，下载与你python版本对应的安装包，cp35对应python3.5， cp37对应python3.7，依次类推，否则也无法安装。