ubuntu16.04 安装CUDA 8.0 和 cuDNN 5.1 /cudnn6.0，可适用于gpu版本的(tensorflow,caffe,mxnet)

转载https://zhuanlan.zhihu.com/p/27890924文章，略有修改，感谢原作者

环境：

• ubuntu 16.04
• GTX 960

安装步骤

• 安装Nvidia驱动

系统设置 --> 软件与更新 --> 附加驱动，如下图选择nvidia官方驱动，然后点击应用更改，安装完成后，重启。

终端输入nvidia-smi显示效果如下图表示安装成功：

• CUDA 8.0安装

sudo chmod +x cuda_8.0.61_375.26_linux.run
sudo sh cuda_8.0.61_375.26_linux.run --tmpdir=/tmp --override

根据命令行提示进行安装：

按`q`退出条款浏览或者按`空格`直到条款末尾，输入 accept 接受条款；

输入n不安装nvidia图像驱动，前面已经安装好驱动；

输入y安装cuda 8.0 toolkit；

回车确认cuda默认安装路径（/usr/local/cuda-8.0）；

输入y安装CUDA 8.0 Samples；

输入CUDA 8.0 Samples安装路径：/home/用户名/CUDA/samples（选择你喜欢的目录进行 安装）。

配置cuda环境变量：

sudo gedit ~/.bashrc

文件的末尾追加下面内容:

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

使环境变量马上生效

source ~/.bashrc

终端输入nvidia-smi显示效果如下图表示安装成功：

• cuDNN 5.1安装

• tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
  sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
  sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
  sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5  
  sudo ln -sf libcudnn.so.5 libcudnn.so  
  sudo ldconfig  

  
  
• cuDnn 6.x 安装

tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
sudo ln -sf libcudnn.so.6.1.10 libcudnn.so.6  
sudo ln -sf libcudnn.so.6 libcudnn.so  
sudo ldconfig  

• 参考链接

深度学习开发环境配置：Ubuntu1 6.04+Nvidia GTX 1080+CUDA 8.0

Ubuntu ldconfig(解决*.so不是符号连接)