安裝前準備

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，因此要設置默認的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設置gcc各版本的優(yōu)先級，優(yōu)先級最高的為系統(tǒng)默認版本，可以用下述命令顯示其優(yōu)先級：

sudo update-alternatives --display gcc

設置默認的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對應的版本（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)文章希望大家以后多多支持腳本之家！

最新評論