安裝前準備
CUDA的主要用途是深度學習，而目前主流的深度學習框架Tensorflow2最高支持CUDA 10.1，因此本文講解在Ubuntu 20.04系統(tǒng)上安裝CUDA 10.1的主要過程。
首先要查看你的NVIDIA顯卡驅(qū)動是否支持cuda10.1版本。
在終端執(zhí)行以下命令：
nvidia-smi
如果 CUDA Version: … 這里的版本號大于等于10.1（我的是10.2），就可以安裝cuda10.1.
關(guān)鍵點：gcc降級
因為Ubuntu20.04自帶的gcc版本為9.3，而cuda10.1不支持gcc-9，因此要手動安裝gcc-7，命令如下：
sudo apt-get install gcc-7 g++-7
安裝完gcc-7，系統(tǒng)中就存在兩個版本的gcc，因此要設(shè)置默認的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)先級，優(yōu)先級最高的為系統(tǒng)默認版本，可以用下述命令顯示其優(yōu)先級：
sudo update-alternatives --display gcc
設(shè)置默認的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)先級：
sudo update-alternatives --display g++
按照官網(wǎng)的安裝步驟下載并安裝CUDA 10.1
CUDA Toolkit 10.1 update2 Archive
如上圖所示，推薦選擇run文件安裝，比較簡潔。
- 當安裝界面啟動，選擇continue。
- 然后輸入accept，回車。
- 由于你的系統(tǒng)中已經(jīng)有了NVIDIA顯卡驅(qū)動，如果不想安裝CUDA 10.1中附帶的驅(qū)動，移動到Driver選項上，按空格鍵將該項取消。如下圖。
移動到Install選項，回車，等待安裝完成。
配置可執(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)站會要求你登陸，如果沒有NVIDIA賬號，注冊后再登陸。
選擇與CUDA 10.1對應(yīng)的版本（7.6.5），點開后選擇 cuDNN Library for Linux，點擊下載。（最好選擇 cuDNN Library for Linux 這個文件格式安裝比較方便）
安裝cuDNN
切換到下載文件所在目錄，解壓下載好的cuDNN壓縮文件到當前目錄：
tar zxvf ./cudnn-10.1-linux-x64-v7.6.5.32.tgz -C ./
將解壓出的
cuda/include/cudnn.h文件復制到/usr/local/cuda/include文件夾
cuda/lib64/下所有文件復制到/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ū)動
到此這篇關(guān)于Ubuntu20.04安裝cuda10.1的步驟(圖文教程)的文章就介紹到這了,更多相關(guān)Ubuntu20.04安裝cuda10.1內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
