Ubuntu18.04安装CUDA10、CUDNN

上篇记录了Ubuntu下安装INVIDIA显卡驱动的方法，尽管可以选择CUDA自带的驱动，但为了避免不必要的问题，尽量单独安装。

如果没有单独安装驱动，建议多找几篇博客，对比来看，了解有哪些坑，才能有效避免入坑。

下文为安装步骤

STEP1：CUDA官网选择适合自己系统的版本下载。

我的系统是Ubuntu18.04、64位，选择CUDA10版本如下：

 STEP2：安装CUDA

在命令行中键入：

sudo sh cuda_10.0.130_410.48_linux.run

一直按Enter直至把声明读完（太多了吧！）

如果驱动是独立安装了，一定要选择不安装驱动！选择如下：

会提示：

***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的时候没有选择安装驱动，提示需要安装驱动，忽略就行。

STEP3：添加环境变量：

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

保存并退出

source ~/.bashrc

STEP4：测试是否安装成功

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

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

结果如下就表明成功啦

STEP5：安装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

完成！