深度学习环境搭建（二）—— Ubutun16.04 安装CUDA、CUDNN

 

        该篇关于安装CUDA、CUDNN的博客是基于已经安装好NVIDA驱动的基础，具体如何安装NVIDIA驱动可参考我的上一篇博客

一、安装CUDA

（1）CUDA官网选择适合自己系统的版本下载。（建议选择CUDA10.0或CUDA10.1，虽说还有很多人在用CUDA9但是CUDA10的趋势越来越主流，很多product都是基于CUDA10运行的）

我的系统是Ubuntu16.04、64位，选择CUDA10版本

（2）安装CUDA（如果是参考我的上一篇博客安装NVIDIA驱动，那么一定采用runfile的方式安装CUDA）

sudo sh cuda_10.0.130_430.64_linux.run

（3）一定要选择不安装驱动，选择如下：（或者不勾选CUDA driver）

会提示：

***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 10.0 functionality to work.

To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:

sudo <CudaInstaller>.run -silent -driver

这是因为我们在安装CUDA的时候没有选择安装驱动，提示需要安装驱动，忽略就行。

（4）添加环境变量（安装CUDA时，会自动在/usr/local目录下生成cuda的文件目录）

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

（5）保存并退出

source ~/.bashrc

（6）测试是否成功

上面的选项选择安装了CUDA例子，运行其中一个来测试是否安装成功：

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

结果如下就表明成功啦

二、安装CUDNN

（1）在官网下载安装包，需要注册登录才能下载。选择适合自己的版本

（2）下载完成后解压并进入文件夹：

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*

（3）在终端查看CUDNN版本：

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

Yes, you have completed the complicated work.