Ubuntu 安装 CUDA 和 cuDNN 详细步骤

1. 确保 nvidia 驱动已安装

先安装 nvidia 驱动，请参考前面文章 https://blog.csdn.net/wohu1104/article/details/107032493

2. 安装 CUDA

2.1 下载安装

在官网 https://developer.nvidia.com/cuda-toolkit-archive如下截图， 选择自己系统对应的 CUDA 版本，

2.2 CUDA 10.2 安装

然后点击进去，进入安装方式页面，如下图
然后在命令行中执行以下语句：

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.runsudo 

sh cuda_10.2.89_440.33.01_linux.run

安装过程如下：

这里注意去勾选 NVIDIA Driver 因为前面已经安装，所以此处不需要再安装。

确认 Yes ，添加动态链接。

安装完成后的输出结果如下：

wohu@ubuntu1804:~/CUDA10-linux$ sudo ./cuda_10.2.89_440.33.01_linux.run 
===========
= Summary =
===========

Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-10.2/
Samples:  Installed in /home/wohu/, but missing recommended libraries

Please make sure that
 -   PATH includes /usr/local/cuda-10.2/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-10.2/lib64, or, add /usr/local/cuda-10.2/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-10.2/bin

Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-10.2/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 440.00 is required for CUDA 10.2 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

Logfile is /var/log/cuda-installer.log

可以看到 /usr/local/ 目录下已经有 cuda -> /usr/local/cuda-10.2// 软链接。

wohu@ubuntu1804:~/CUDA10-linux$ ll /usr/local/
total 48
drwxr-xr-x 12 root root 4096 6月  30 09:25 ./
drwxr-xr-x 11 root root 4096 2月   4 02:25 ../
drwxr-xr-x  2 root root 4096 2月   4 02:22 bin/
lrwxrwxrwx  1 root root   21 6月  30 09:25 cuda -> /usr/local/cuda-10.2//
drwxr-xr-x 18 root root 4096 6月  30 09:25 cuda-10.2/
drwxr-xr-x  2 root root 4096 2月   4 02:22 etc/
drwxr-xr-x  2 root root 4096 2月   4 02:22 games/
drwxr-xr-x  2 root root 4096 2月   4 02:22 include/
drwxr-xr-x  4 root root 4096 6月  29 18:35 lib/
lrwxrwxrwx  1 root root    9 6月  16 16:24 man -> share/man/
drwxr-xr-x  2 root root 4096 2月   4 02:22 sbin/
drwxr-xr-x  6 root root 4096 2月   4 02:25 share/
drwxr-xr-x  2 root root 4096 2月   4 02:22 src/
drwxr-xr-x  6 root root 4096 6月  19 18:49 sunlogin/
wohu@ubuntu1804:~/CUDA10-linux$ 

2.3 CUDA 10 安装

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n

Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-10.0 ]: 回车

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: y

Enter CUDA Samples Location
 [ default is /home/linkdome ]: 回车

2.4 添加环境变量

打开 ~/.bashrc

sudo vi ~/.bashrc

在文件最后添加以下环境变量到 ~/.bashrc 并保存退出。

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

使其生效

sudo source  ~/.bashrc

2.5 验证安装成功

最后输入命令 nvcc -V 有如下输出，说明已安装成功。

wohu@ubuntu1804:/$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
wohu@ubuntu1804:/$

3. 安装 cuDNN

3.1 下载

官网地址 https://developer.nvidia.com/rdp/form/cudnn-download-survey 注册账号，并下载对应版本的 cuDNN 版本。

3.2 安装

对压缩包进行解压：

tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

解压过程如下：

wohu@ubuntu1804:~/CUDA10-linux$ tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz 
cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.6.5
cuda/lib64/libcudnn_static.a
wohu@ubuntu1804:~/CUDA10-linux$ 

执行以下命令：

sudo cp cuda/include/cudnn.h	/usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn*	/usr/local/cuda/lib64/

3.3 验证是否安装成功

输入以下命令，

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

有如下输出，说明已安装成功。