安裝前準(zhǔn)備
CUDA的主要用途是深度學(xué)習(xí)，而目前主流的深度學(xué)習(xí)框架Tensorflow2最高支持CUDA 10.1，因此本文講解在Ubuntu 20.04系統(tǒng)上安裝CUDA 10.1的主要過程。
首先要查看你的NVIDIA顯卡驅(qū)動(dòng)是否支持cuda10.1版本。
在終端執(zhí)行以下命令：
nvidia-smi
如果 CUDA Version: … 這里的版本號(hào)大于等于10.1（我的是10.2），就可以安裝cuda10.1.
關(guān)鍵點(diǎn)：gcc降級(jí)
因?yàn)閁buntu20.04自帶的gcc版本為9.3，而cuda10.1不支持gcc-9，因此要手動(dòng)安裝gcc-7，命令如下：
sudo apt-get install gcc-7 g++-7
安裝完gcc-7，系統(tǒng)中就存在兩個(gè)版本的gcc，因此要設(shè)置默認(rèn)的gcc，命令如下：
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
此命令可以通過update-alternatives設(shè)置gcc各版本的優(yōu)先級(jí)，優(yōu)先級(jí)最高的為系統(tǒng)默認(rèn)版本，可以用下述命令顯示其優(yōu)先級(jí)：
sudo update-alternatives --display gcc
設(shè)置默認(rèn)的g++也是如此：
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
顯示g++優(yōu)先級(jí)：
sudo update-alternatives --display g++
按照官網(wǎng)的安裝步驟下載并安裝CUDA 10.1
CUDA Toolkit 10.1 update2 Archive
如上圖所示，推薦選擇run文件安裝，比較簡(jiǎn)潔。
- 當(dāng)安裝界面啟動(dòng)，選擇continue。
- 然后輸入accept，回車。
- 由于你的系統(tǒng)中已經(jīng)有了NVIDIA顯卡驅(qū)動(dòng)，如果不想安裝CUDA 10.1中附帶的驅(qū)動(dòng)，移動(dòng)到Driver選項(xiàng)上，按空格鍵將該項(xiàng)取消。如下圖。
移動(dòng)到Install選項(xiàng)，回車，等待安裝完成。
配置可執(zhí)行文件及庫路徑
安裝完成后，需要為CUDA 10.1配置環(huán)境。
首先打開家目錄下的.bashrc文件： gedit ~/.bashrc
在文件的末尾添加如下內(nèi)容：
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并關(guān)閉文件，完成配置。
更新環(huán)境：source ~/.bashrc
安裝cuDNN
下載cuDNN
Download cuDNN
打開鏈接后，網(wǎng)站會(huì)要求你登陸，如果沒有NVIDIA賬號(hào)，注冊(cè)后再登陸。
選擇與CUDA 10.1對(duì)應(yīng)的版本（7.6.5），點(diǎn)開后選擇 cuDNN Library for Linux，點(diǎn)擊下載。（最好選擇 cuDNN Library for Linux 這個(gè)文件格式安裝比較方便）
安裝cuDNN
切換到下載文件所在目錄，解壓下載好的cuDNN壓縮文件到當(dāng)前目錄：
tar zxvf ./cudnn-10.1-linux-x64-v7.6.5.32.tgz -C ./
將解壓出的
cuda/include/cudnn.h文件復(fù)制到/usr/local/cuda/include文件夾
cuda/lib64/下所有文件復(fù)制到/usr/local/cuda/lib64文件夾
為上述文件添加讀取和執(zhí)行權(quán)限：
sudo chmod 755 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安裝完畢
參考文章:
Ubuntu 20.04 CUDA&cuDNN安裝方法
Ubuntu：安裝 cuda10.1 驅(qū)動(dòng)
到此這篇關(guān)于Ubuntu20.04安裝cuda10.1的步驟(圖文教程)的文章就介紹到這了,更多相關(guān)Ubuntu20.04安裝cuda10.1內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
