1. 確保 nvidia 驅(qū)動已安裝

先安裝 nvidia 驅(qū)動，請參考前面文章 https://blog.csdn.net/wohu1104/article/details/107032493

2. 安裝 CUDA

2.1 下載安裝

在官網(wǎng) https://developer.nvidia.com/cuda-toolkit-archive如下截圖， 選擇自己系統(tǒng)對應(yīng)的 CUDA 版本，

2.2 CUDA 10.2 安裝

然后點擊進去，進入安裝方式頁面，如下圖
然后在命令行中執(zhí)行以下語句：

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 因為前面已經(jīng)安裝，所以此處不需要再安裝。

確認 Yes ，添加動態(tài)鏈接。

安裝完成后的輸出結(jié)果如下：

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 librariesPlease 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 rootTo uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-10.2/binPlease 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 --driverLogfile is /var/log/cuda-installer.log

可以看到 /usr/local/ 目錄下已經(jīng)有 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: acceptInstall NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: nInstall the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: yEnter 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: yInstall the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: yEnter CUDA Samples Location[ default is /home/linkdome ]: 回車

2.4 添加環(huán)境變量

打開 ~/.bashrc

sudo vi ~/.bashrc

在文件最后添加以下環(huán)境變量到 ~/.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 下載

官網(wǎng)地址 https://developer.nvidia.com/rdp/form/cudnn-download-survey 注冊賬號，并下載對應(yīng)版本的 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$ 

執(zhí)行以下命令：

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

有如下輸出，說明已安裝成功。

總結(jié)

以上是生活随笔為你收集整理的Ubuntu 安装 CUDA 和 cuDNN 详细步骤的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。