pytorch-GPU版本安装

1、安装过程

1.1 首先安装驱动

1.1.1 先卸载原有驱动

sudo apt-get remove --purge nvidia*

1.1.2禁用nouveau驱动

sudo gedit /etc/modprobe.d/blacklist.conf

    在文件最后添加：

blacklist nouveau
options nouveau modeset=0

   然后执行

sudo update-initramfs -u

   重启后执行

lsmod | grep nouveau

   如果终端没有任何输出，说明禁用成功。

1.1.3 下载并安装驱动

   nvidia官网下载

   关闭图像界面

sudo service lightdm stop

    ctrl+alt+F1进入命令行（ctrl+alt+F5也有可能）

    cd到驱动文件所在的目录，给run文件赋予执行权限并安装

sudo chmod a+x NVIDIA-Linux-x86_64-？？？.run
sudo ./NVIDIA-Linux-x86_64-？？？.run –no-opengl-files

1.1.4测试

nvidia-smi

1.2 安装cuda和cudnn

  1.2.1我安装的是cuda9.0+cudnn7.0.5

sudo sh cuda_9.0.176_384.81_linux.run 

    接受协议后不要再重新安装驱动，accept  n y y y ...

    设置环境变量

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

   测试用nvcc -V

   解压cudnn

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

   查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

  安装pytorch

  在anaconda下安装，建议新建一个环境

conda create -n pytorch python=3.6.0
source activate pytorch
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision

  import torch成功

  说明安装完成

2、总结

      简单记录。。