安裝前準備

CUDA的主要用途是深度學習，而目前主流的深度學習框架Tensorflow2最高支持CUDA 10.1，因此本文講解在Ubuntu 20.04系統上安裝CUDA 10.1的主要過程。

首先要查看你的NVIDIA顯卡驅動是否支持cuda10.1版本。
在終端執行以下命令：
nvidia-smi

如果 CUDA Version: … 這里的版本號大于等于10.1（我的是10.2），就可以安裝cuda10.1.

關鍵點：gcc降級

因為Ubuntu20.04自帶的gcc版本為9.3，而cuda10.1不支持gcc-9，因此要手動安裝gcc-7，命令如下：

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

安裝完gcc-7，系統中就存在兩個版本的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各版本的優先級，優先級最高的為系統默認版本，可以用下述命令顯示其優先級：

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++優先級：

sudo update-alternatives --display g++

按照官網的安裝步驟下載并安裝CUDA 10.1

CUDA Toolkit 10.1 update2 Archive

如上圖所示，推薦選擇run文件安裝，比較簡潔。

• 當安裝界面啟動，選擇continue。
• 然后輸入accept，回車。
• 由于你的系統中已經有了NVIDIA顯卡驅動，如果不想安裝CUDA 10.1中附帶的驅動，移動到Driver選項上，按空格鍵將該項取消。如下圖。
  

移動到Install選項，回車，等待安裝完成。

配置可執行文件及庫路徑

安裝完成后，需要為CUDA 10.1配置環境。

首先打開家目錄下的.bashrc文件： gedit ~/.bashrc

在文件的末尾添加如下內容：

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}}

保存并關閉文件，完成配置。

更新環境：source ~/.bashrc

安裝cuDNN

下載cuDNN

Download cuDNN

打開鏈接后，網站會要求你登陸，如果沒有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文件夾

為上述文件添加讀取和執行權限：

sudo chmod 755 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安裝完畢

參考文章:

Ubuntu 20.04 CUDA&cuDNN安裝方法
Ubuntu：安裝 cuda10.1 驅動

到此這篇關于Ubuntu20.04安裝cuda10.1的步驟(圖文教程)的文章就介紹到這了,更多相關Ubuntu20.04安裝cuda10.1內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家！