一、安裝Nvidia顯卡
參考文獻(xiàn)：從零到一保姆級(jí)Ubuntu深度學(xué)習(xí)服務(wù)器環(huán)境配置教程
看文獻(xiàn)中“三、 NVIDIA驅(qū)動(dòng)安裝”
安裝NVIDIA驅(qū)動(dòng)，這也是安裝CUDA10.0及其對(duì)應(yīng)版本的CuDNN和tensorflow的重要步驟。
1.1 輸入顯卡型號(hào)查看支持顯卡驅(qū)動(dòng)的版本
1.1.1 英偉達(dá)中國(guó)驅(qū)動(dòng)官網(wǎng)
進(jìn)入英偉達(dá)中國(guó)驅(qū)動(dòng)官網(wǎng)
1.1.2 輸入顯卡型號(hào)查詢
1.1.3 查看搜索結(jié)果
1.2 下載安裝Nvidia
1.2.1 方法一
直接在Ubuntu系統(tǒng)中的soft updates中安裝525-server（專有）版本
注意：
1.2.1 方法二
自己試一下，我沒(méi)用下面的方法。
參考文獻(xiàn)：在ubuntu安裝nvidia驅(qū)動(dòng) (親測(cè)有效，這是方法二)
參考文獻(xiàn)：在ubuntu安裝nvidia驅(qū)動(dòng)（親測(cè)有效，這是方法一）)
二、安裝CUDA11.6.0
參考文獻(xiàn)：ubuntu18.04安裝nvidia_driver_510+cuda_11.6+cudnn_11.x
參考文獻(xiàn)：從零到一保姆級(jí)Ubuntu深度學(xué)習(xí)服務(wù)器環(huán)境配置教程
2.1 檢測(cè)自己電腦GPU是否兼容CUDA（N卡支持）
lspci | grep -i nvidia
顯示compatible兼容
2.2 進(jìn)入CUDA官網(wǎng)
CUDA官網(wǎng):https://developer.nvidia.com/cuda-toolkit-archive
2.3 下載安裝CUDA11.6.0
我比較習(xí)慣把一些下載安裝同一個(gè)文件夾“software”
第一步：home 下新建文件
cd # 返回home目錄
mkdir software # 新建 software文件夾
第二步：后續(xù)在software文件夾下載的CUDA11.6.0安裝包
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
第三步（可省略）：給run文件權(quán)限
sudo chmod 777 cuda_11.6.0_510.39.01_linux.run
第四步：運(yùn)行run文件（–override：表示覆蓋替換文件意思）
sudo sh cuda_11.6.0_510.39.01_linux.run --override
等待后續(xù)：
（1）選擇continue
(2)輸入：accept
(3)選項(xiàng)條款選擇
“Enter”按鍵選中或者取消
前面已經(jīng)安裝過(guò)Nvidia顯卡525，此處不需要Driver（前兩個(gè)都取消）（X代表選中）
最后，Install，等待完成
2.4 安裝CUDA11.6.0后的配置
第一步：配置相關(guān)環(huán)境變量。
（1）打開~/.bashrc文件，代碼如下：
sudo vim ~/.bashrc
（2）將以下內(nèi)容寫入到~/.bashrc尾部，并保存成功。
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
（3）保存，退出。終端執(zhí)行
source ~/.bashrc
后續(xù)可根據(jù)意愿進(jìn)行，我也不確定。
參考文獻(xiàn)：ubuntu18.04安裝nvidia_driver_510+cuda_11.6+cudnn_11.x
第二步：利用vim命令打開在/etc/profile文件中，添加CUDA環(huán)境變量，代碼如下：
(1)
sudo vim /etc/profile
(2)打開文檔都在文檔結(jié)尾加上下面兩句：
PATH=/usr/local/cuda/bin:$PATH
export PATH
(3)保存后, 執(zhí)行下列命令, 使環(huán)境變量立即生效，代碼如下：
source /etc/profile
第三步：同時(shí)添加lib庫(kù)路徑，在 /etc/ld.so.conf.d/新建文件 cuda.conf，命令如下：
（1）
sudo vim /etc/ld.so.conf.d/cuda.conf
（2）在文中加入下面內(nèi)容：
/usr/local/cuda/lib64
（3）執(zhí)行下列命令使之立刻生效，代碼如下：
sudo ldconfig
2.5 利用測(cè)試CUDA的samples來(lái)測(cè)試cuda安裝是否成功
cd /usr/local/cuda-11.6/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
一般都會(huì)面臨如下問(wèn)題：
11.6版本cuda的安裝目錄/usr/local/cuda-11.6/samples里只有一個(gè)txt文件，大致意思是告訴你新版本的cuda，samples中內(nèi)容需要自己在github下載。
解決方法：
第一步：下載測(cè)試示例
（1）切換到/usr/local/cuda-11.6/samples目錄下
cd /usr/local/cuda-11.6/samples/
（2）github下載
git clone https://gitee.com/liwuhaoooo/cuda-samples.git
大概率無(wú)權(quán)訪問(wèn)，此時(shí)在cuda-11.6文件夾下打開終端
su
輸入密碼切換超級(jí)用戶
chmod 777 samples
再次執(zhí)行g(shù)it clone 就可以了。
第二步：找到1_Utilities/deviceQuery文件夾
（1）切換到deviceQuery文件夾下
（2）運(yùn)行
sudo make
./deviceQuery
（3）顯示pass
注意：
三、安裝cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x
參考文獻(xiàn)：從零到一保姆級(jí)Ubuntu深度學(xué)習(xí)服務(wù)器環(huán)境配置教程
3.1 打開官網(wǎng)使用郵箱注冊(cè)（我用的qq郵箱）
看參考文獻(xiàn)（cuDNN部分）：從零到一保姆級(jí)Ubuntu深度學(xué)習(xí)服務(wù)器環(huán)境配置教程
3.2 下載 cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x
同樣，下載之后放在software文件夾中。
3.3 安裝 cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x
參考文獻(xiàn)：Ubuntu20.04安裝Nvidia顯卡驅(qū)動(dòng)、CUDA11.5、cuDNN8.3、Anaconda及Tensorflow-GPU版本詳細(xì)圖文操作教程
第一步：查看官方安裝方法
Nvidia官方安裝教程：https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
完全按照官方操作即可，大概意思是將文檔中的X.Y修改成CUDA版本號(hào)、v8.x.x.x修改成cuDNN版本號(hào)，這里我們將X.Y改為11.5、v8.x.x.x修改為8.3.2.44，同學(xué)們參照修改就可以了。
第二步：安裝
（1）切換到下載路徑
cd ~/software
（2）解壓
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
（3）成功提取文件后不用管它，直接執(zhí)行下面命令（不用修改，沒(méi)有版本號(hào)之類）文章來(lái)源：http://www.zghlxwxcb.cn/news/detail-445109.html
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
3.4 驗(yàn)證安裝
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-445109.html
到了這里，關(guān)于Ubuntu 20.04 RTX 4090顯卡 深度學(xué)習(xí)環(huán)境配置（Nvidia顯卡驅(qū)動(dòng)、CUDA11.6.0、cuDNN8.5）的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
