Ubuntu 20.04 CUDA&cuDNN安裝方法(圖文教程)

CUDA安裝下載cuda

輸入nvidia-smi命令查看支持的cuda版本

如果無(wú)法查看，則說(shuō)明尚未安裝nvidia驅(qū)動(dòng)，點(diǎn)擊附加驅(qū)動(dòng)，選擇對(duì)應(yīng)版本的驅(qū)動(dòng)即可自動(dòng)下載。

從https://developer.nvidia.com/cuda-toolkit-archive下載對(duì)應(yīng)版本的cuda，選擇18.04的版本即可。

gcc降級(jí)

Ubuntu20.04自帶的gcc版本為9.7.0，需要添加gcc7才可安裝cuda10.2，輸入命令安裝gcc7

apt-get install gcc-7 g++-7

查看gcc版本，可以看到目前系統(tǒng)中存在7和9兩個(gè)版本

使用update-alternatives進(jìn)行版本切換，輸入以下命令：

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

此時(shí)輸入sudo update-alternatives --config gcc命令查看gcc的默認(rèn)版本，可以看到當(dāng)前默認(rèn)gcc版本為7，即切換成功。

安裝cuda

輸入命令進(jìn)行安裝

sudo sh cuda_10.2.89_440.33.01_linux.run

點(diǎn)擊continue

輸入accept

cuda安裝包是自帶顯卡驅(qū)動(dòng)的，所以這一步按空格去掉安裝顯卡驅(qū)動(dòng)的選項(xiàng)，然后選擇install

配置環(huán)境變量

輸入gedit ~/.bashrc命令打開(kāi)文件，在文件結(jié)尾輸入以下語(yǔ)句，保存。

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下載相應(yīng)版本的cudnn，需要登陸。

下載解壓之后，將cuda/include/cudnn.h文件復(fù)制到usr/local/cuda/include文件夾，將cuda/lib64/下所有文件復(fù)制到/usr/local/cuda/lib64文件夾中，并添加讀取權(quán)限：

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

至此cuDNN安裝完成

到此這篇關(guān)于Ubuntu 20.04 CUDA&cuDNN安裝方法(圖文教程)的文章就介紹到這了,更多相關(guān)Ubuntu20.04 CUDA&cuDNN安裝 內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！