Ubuntu 20.04 CUDA&cuDNN安裝方法(圖文教程)
CUDA安裝下載cuda
輸入nvidia-smi命令查看支持的cuda版本

如果無法查看,則說明尚未安裝nvidia驅(qū)動,點擊附加驅(qū)動,選擇對應版本的驅(qū)動即可自動下載。

從https://developer.nvidia.com/cuda-toolkit-archive下載對應版本的cuda,選擇18.04的版本即可。

gcc降級
Ubuntu20.04自帶的gcc版本為9.7.0,需要添加gcc7才可安裝cuda10.2,輸入命令安裝gcc7
apt-get install gcc-7 g++-7
查看gcc版本,可以看到目前系統(tǒng)中存在7和9兩個版本

使用update-alternatives進行版本切換,輸入以下命令:
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100 sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
此時輸入sudo update-alternatives --config gcc命令查看gcc的默認版本,可以看到當前默認gcc版本為7,即切換成功。

安裝cuda
輸入命令進行安裝
sudo sh cuda_10.2.89_440.33.01_linux.run

點擊continue

輸入accept

cuda安裝包是自帶顯卡驅(qū)動的,所以這一步按空格去掉安裝顯卡驅(qū)動的選項,然后選擇install
配置環(huán)境變量
輸入gedit ~/.bashrc命令打開文件,在文件結(jié)尾輸入以下語句,保存。
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
更新環(huán)境變量配置
source ~/.bashrc
至此cuda安裝完成,輸入nvcc -V命令查看cuda信息

安裝cuDNN
從https://developer.nvidia.com/cudnn下載相應版本的cudnn,需要登陸。

下載解壓之后,將cuda/include/cudnn.h文件復制到usr/local/cuda/include文件夾,將cuda/lib64/下所有文件復制到/usr/local/cuda/lib64文件夾中,并添加讀取權(quán)限:
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
至此cuDNN安裝完成
到此這篇關于Ubuntu 20.04 CUDA&cuDNN安裝方法(圖文教程)的文章就介紹到這了,更多相關Ubuntu20.04 CUDA&cuDNN安裝 內(nèi)容請搜索本站以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持本站!
版權(quán)聲明:本站文章來源標注為YINGSOO的內(nèi)容版權(quán)均為本站所有,歡迎引用、轉(zhuǎn)載,請保持原文完整并注明來源及原文鏈接。禁止復制或仿造本網(wǎng)站,禁止在非maisonbaluchon.cn所屬的服務器上建立鏡像,否則將依法追究法律責任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來,僅供學習參考,不代表本站立場,如有內(nèi)容涉嫌侵權(quán),請聯(lián)系alex-e#qq.com處理。
關注官方微信