Ubuntu1604+GTX1080+CUDA8.0+CUDNN5.1+Pytorch 安装说明

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/caicaiatnbu/article/details/86505121

GPU驱动部分，CUDA8.0 的安装和CUDNN5.1 的配置请参考 Ubuntu1604+GTX1060+CUDA8.0+CUDNN5.1+Caffe 安装说明 一文中前4步（包含第4布），在对NVIDIA_CUDA-8.0_Samples文件夹进行make操作可能报错如下：

make[1]: Leaving directory '/home/ubuntu/NVIDIA_CUDA-8.0_Samples/3_Imaging/imageDenoising'

但是这并不会影响我们运行./deviceQuery,如下图所示：

• 安装Pytorch， 这里我们选择的python版本为3.5 , 这里我们参考了 各版本Pytorch安装详解 一文

sudo pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
sudo pip3 install torchvision

但是发现报错  TypeError: unsupported operand type(s) for -=: 'Retry' and 'int'，如下图所示：

 产生这个bug的原因是python2.7版本pip模块的一个bug，需要升级pip来解决这个问题。

解决办法，在终端输入以下命令：

sudo apt-get remove python-pip python3-pip
sudo wget https://bootstrap.pypa.io/get-pip.py
sudo python get-pip.py
sudo python3 get-pip.py

• 测试是否成功安装