Ubuntu下安装cuda+cudnn

ubuntu下安装cuda+cudnn

---

第一步 安装nvida驱动

安装nvidia驱动，请参考我另一篇教程Ubuntu下安装nvidia显卡驱动

第二步 安装cuda8.0

1、下载cuda

进入 cuda8.0 download ，选择合适的 CUDA 类型下载。

如上图所示我选择的是linux x86_64 ubuntu 16.04环境，最后一个选项是安装包形式，我选择的是runfile安装包，直接运行

点击downlad即可下载

2、安装cuda相关依赖

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

3、安装cuda

1. Ctrl+Alt+F1进入非图形界面，关闭终端 sudo service lightdm stop
2. 安装
   
cd /home/lihailin/Downloads
run ./cuda_8.0.61_375.26_linux.run

   a）会先有个阅读声明，一直按q即可，然后accept
   b）第一个选项install nvidia accelerated Graphics Driver （n,因为我们之前安装过了nvidia显卡）
   c）第二个选项install the OpenGL libraries 双显卡（指的是集成显卡+独立显卡）选n，单卡（如果没有集成显卡，只有一个或多个支持GPU的显卡）选y。双显卡选y的话，会出现黑屏或登录界面无限循环的问题。

3. 若第三步无警告和报错，出现下列显示，即安装暂时成功。
   

   
   Driver ：Installed
   Toolkit ：Installed in /usr/local/cuda-8.5
   Samples ：Installed in /home/lihailin

4. 重新启动图形化界面 sudo service lightdm start， Alt + ctrl +F7，返回到图形化登录界面，输入密码登录。
   如果能够成功登录，则表示不会遇到循环登录的问题，基本说明cuda的安装成功了。

5. 搭配环境
   终端输入

echo 'export PATH=/usr/local/cuda-8.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

4、验证cuda是否安装成功

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

若看到类似以下信息则说明 cuda 已安装成功： 

cd ../bandwithTest
sudo make
./bandwithTest

若看到类似以下信息则说明 cuda 已安装成功： 

第三步 安装cudnn

1. 下载cudnn5.1 cudnn download
2. 解压并复制到/usr/local
   
cd ~/lihailin/Downloads
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
sudo ldconfig


---

环境

1. GTX970M，nvidia版本nvidia-384
2. ubuntu 16.04
3. 安装的 cuda8.0 + cudnn5.1

参考资料

1. 二、CUDA安装和测试
2. Ubuntu16.04 Caffe 安装步骤记录 部分可参考