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 with the name of this run file:

sudo .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

 

完成！