Ubuntu16.04 安装cuda9.0+cudnn7.05 + Anaconda 3.0 + pytorch gpu

Ubuntu16.04 安装cuda9.0+cudnn7.05 + Anaconda 3.0 + pytorch gpu@TOC

Cuda+Cudnn

我的配置是Ubuntu16.04，双路1080ti显卡。
1.禁用nouveau
ubuntu 16.04默认安装了第三方开源的驱动程序nouveau，安装nvidia显卡驱动首先需要禁用nouveau，不然会碰到冲突的问题，导致无法安装nvidia显卡驱动：

编辑文件 blacklist.conf
sudo vim /etc/modprobe.d/blacklist.conf
文件最后插入
sudo vim /etc/modprobe.d/blacklist.conf
更新
sudo update-initramfs -u
检测是否禁用
lsmod | grep nouveau

2.安装NVIDIA驱动

按 ctrl+alt+f1，进入ttf1，停止lightdm
sudo service lightdm stop
如果之前安装NVIDIA驱动，先卸载
sudo apt-get purge nvidia-*
检查卸载是否成功（成功则没有任何输出）
nvidia-smi
下载NVIDIA驱动，进行安装即可
sudo ./ NVIDIA-Linux-x86_64-（此处对应版本号）.run 

3.安装cuda9.0+cudnn7.0.5
官网下载对应版本的cuda和cudnn
下载linux版本即可，不需要unbuntu版本
cuda:

赋予cuda执行权限
chmod +x cuda_（版本号）_linux.run
执行安装
./cuda_（版本号）_linux.run –no-opengl-libs 

安装过程中一些选择：
accept
n（不要安装driver）
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
//运行
source ~/.bashrc
//检查安装情况，若出现几行版本型号，则安装成功：
nvcc --version

cudnn:

//解压
tar -xvf cudnn-（版本号）-linux-x64-v7.tgz
//安装，其实就是拷贝几个文件
cd cuda 
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*
//检查安装情况，若出现几行版本型号，则安装成功：
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

4.anaconda
https://blog.youkuaiyun.com/u012243626/article/details/82469174
https://blog.youkuaiyun.com/hello_dear_you/article/details/84573500
添加镜像：
https://blog.youkuaiyun.com/qq_33039859/article/details/81328805

5.pytorch

//创建环境（可以通过此方法创建多个环境，pytorch.tensorflow等等）
conda create -n 文件名 python=版本号
//安装pytorch 具体参考地址：https://pytorch.org/get-started/locally/
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

最近遇到的问题：
服务器双cuda版本：https://blog.youkuaiyun.com/yinxingtianxia/article/details/80462892