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
- 测试是否成功安装