一、NVIDIA顯卡驅(qū)動安裝
nvidia顯卡驅(qū)動安裝方式有三種:使用ubuntu附加驅(qū)動的方式;使用命令行方式安裝;使用.run文件的方式進行安裝,
1.1 ubuntu附加驅(qū)動的方式
點擊菜單中的Additional Drivers選擇適合的驅(qū)動版本進行安裝,該方法最方便快捷(但有時會翻車)
1.2 命令行方式安裝
更新所有的軟件包
sudo add-apt-repository ppa:graphics-drivers/ppa # 加入官方ppa源
sudo apt update # 檢查軟件包更新列表
apt list --upgradable # 查看可更新的軟件包列表
sudo apt upgrade # 更新所有可更新的軟件包
安裝顯卡驅(qū)動
ubuntu-drivers devices # ubuntu檢測n卡的可選驅(qū)動
sudo apt install nvidia-driver-510 # 根據(jù)自己的n卡可選驅(qū)動下載顯卡驅(qū)動
1.3 .run文件方式安裝
詳見我的另一篇博客Ubunut20.04/22.04安裝NVIDIA驅(qū)動
該方法操作步驟最麻煩(最不容易翻車)
1.4 雙顯卡驅(qū)動選擇
如果有兩張不同品牌獨顯,需要選擇nvidia顯卡,有兩種方法,具體如下:
1.可以打開終端使用nvidia-settings
命令選擇,點擊Prime profiles 選擇NVIDIA (Performance Mode)
調(diào)整后重啟系統(tǒng)
2.也可以使用prime-select命令
sudo prime-select query //查看當前使用顯卡
sudo prime-select nvidia //使用nvidia顯卡
sudo prime-select intel //使用intel顯卡
使用sudo prime-select nvidia
命令后可以使用reboot
重啟系統(tǒng)
1.4 驗證NVIDIA驅(qū)動安裝情況
可以使用以下命令查看是否加載了nvidia驅(qū)動
sudo nvidia-settings # 更改Nvidia驅(qū)動設置
nvidia-smi # 查看顯卡基本信息
如果有問題的話,可以gpu-manager查看一下原因,主要可以查看與下圖紅框中是否一致,如果有問題,具體問題具體分析
sudo gpu-manager
二、CUDA安裝
2.1 查看版本對應關系
首先需要去NVIDIA CUDA Toolkit Release Notes中查看自己的顯卡驅(qū)動對應的CUDA版本,下圖為版本對應關系:
查看nvidia-smi也可以得到驅(qū)動版本,可以通過該版本號查詢上面的表格,也可以在這個界面里得到該驅(qū)動程序版本對應的最高cuda版本,即CUDA Version 11.7
2.2 下載CUDA
cuda官網(wǎng)網(wǎng)址為CUDA Toolkit Archive
這里我選擇了11.7版本的CUDA進行安裝,可以安裝你所需要配置的版本,最高支持到nvidia-smi中輸出的版本
根據(jù)自己系統(tǒng)情況選擇相應選項,最后一個選擇runfile最不容易出問題
使用如下命令下載該文件
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
下載完成
2.3 CUDA安裝
然后執(zhí)行該步驟進行安裝
sudo sh cuda_11.7.0_515.43.04_linux.run
選擇continue繼續(xù)
輸入accept,回車
第一個是選擇驅(qū)動,給他回車取消,因為我們已經(jīng)有安裝驅(qū)動了,然后移動到install安裝
出現(xiàn)以下summary時安裝完成然后在.bashrc里配置環(huán)境變量
sudo gedit ~/.bashrc
在打開文件的最后一行之后加上以下字段:
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
然后刷新環(huán)境變量
source ~/.bashrc
使用以下命令查看CUDA安裝情況
nvcc -V
三、安裝cudnn
這個要是跑深度學習必須裝,不需要不裝!
3.1 下載安裝
去官網(wǎng)下載:https://developer.nvidia.com/rdp/cudnn-download
選擇Local Installer for Linux x86_64 (Tar)下載
下載后解壓進入該目錄拷貝相關文件
cd cudnn-linux-x86_64-8.6.0.163_cuda11-archive
sudo cp include/cudnn*.h /usr/local/cuda/include
sudo cp -p lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
3.2 查看安裝版本
舊版本cuDNN使用以下命令查看版本號:文章來源:http://www.zghlxwxcb.cn/news/detail-814124.html
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
而高版本的cuDNN的版本號已經(jīng)不在cudnn.h中了, 而是在cudnn_version.h中,我們也需要將cudnn_version.h拷貝到 /usr/local/cuda/include中,然后再使用以下命令查看:文章來源地址http://www.zghlxwxcb.cn/news/detail-814124.html
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
到了這里,關于【ubuntu環(huán)境配置】超詳細ubuntu20.04/22.04安裝nvidia驅(qū)動/CUDA/cudnn的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!