參考：
https://zhuanlan.zhihu.com/p/59618999
https://blog.csdn.net/linhai1028/article/details/79445722/
Windows+Ubuntu從雙系統(tǒng)安裝到CUDA cuDNN docker 配置K21

https://blog.csdn.net/qq_45831128/article/details/127060475

https://blog.csdn.net/kunhe0512/article/details/125061911

本機(jī)環(huán)境

Ubuntu 20.04
英偉達(dá) RTX 3050 Ti 顯卡

1 相關(guān)查詢命令

1）Ubuntu下查看Nvidia顯卡的詳細(xì)信息：

nvidia-smi

2）查看cuda 版本：

nvcc -V

3）查看cudnn版本：

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

4）在終端輸入命令，實(shí)時(shí)查看GPU的使用情況：

watch -n 1 nvidia-smi

5）在Ubuntu下查詢顯卡型號(hào)

nvidia-smi -a |grep -i Product\ Name

一、Ubuntu 20.04 安裝NVIDIA顯卡驅(qū)動(dòng)

下載 NVIDIA-Linux-x86_64-515.57.run
這個(gè)版本驅(qū)動(dòng)比較穩(wěn)定。

二、Ubuntu 20.04 安裝cuda 11.7

1、安裝CUDA 11.7

cuda 下載地址：https://developer.nvidia.com/zh-cn/cuda-downloads

第二步就是安裝CUDA。這里我們安裝CUDA最新的11.7版本。進(jìn)入Nvidia cuda下載地址：
Nvidia cuda11.7 下載地址

我們這里選擇的Ubuntu 20.04下的deb安裝類型。

然后依次執(zhí)行下述官方提供的幾條命令：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

或者用非root用戶安裝
參考: 非root用戶安裝cuda與cudnn

3、配置CUDA環(huán)境變量

安裝完成之后，加入到環(huán)境變量，可用gedit

gedit ~/.bashrc

在.bashrc末尾添加兩行環(huán)境變量

export PATH=$PATH:/usr/local/cuda-11.7/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  #根據(jù)CUDA版本更換路徑

最后，使環(huán)境變量生效

source ~/.bashrc

測(cè)試

nvcc -V

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_18:49:52_PDT_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0

三、cudnn8.4 安裝

cudnn下載地址：https://developer.nvidia.com/rdp/cudnn-archive

需要提前注冊(cè)，找到對(duì)應(yīng)版本，下載cuDNN Library for Linux壓縮包即可

解壓壓縮包，并終端進(jìn)入文件夾，執(zhí)行以下

#進(jìn)入解壓目錄
cd cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive
# 復(fù)制cudnn頭文件
sudo cp ./include/* /usr/local/cuda-11.7/include/  
# 復(fù)制cudnn的庫(kù)
sudo cp ./lib/* /usr/local/cuda-11.7/lib64/ 

#給予權(quán)限
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h 
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*

（3）測(cè)試

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

文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-409326.html

到了這里，關(guān)于Ubuntu 20.04 安裝NVIDIA顯卡驅(qū)動(dòng)+cuda 11.7+cudnn 8.4的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！