安裝前准備
CUDA的主要用途是深度學習，而目前主流的深度學習框架Tensorflow2最高支持CUDA 10.1，因此本文講解在Ubuntu 20.04系統上安裝CUDA 10.1的主要過程。
首先要安裝NVIDIA顯卡驅動，這個可以直接在設置附加驅動里選擇對應版本系統會自動安裝
也可選擇手動安裝，download Nvidia-driver
然后查看你的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
download cuda
如上圖所示，推薦選擇run文件安裝，比較簡潔。
- 當安裝界面啟動，選擇continue。
- 然后輸入accept，回車。
- 由於你的系統中已經有了NVIDIA顯卡驅動，如果不想安裝CUDA 10.1中附帶的驅動，移動到Driver選項上，按空格鍵將該項取消。如下圖。
- 移動到Install選項，回車，等待安裝完成。
配置可執行文件及庫路徑
安裝完成后，需要為CUDA 10.1配置環境。
- 首先打開家目錄下的
.bashrc文件(如果使用zsh終端，則打開.zshrc)：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
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*
*文章轉自https://blog.csdn.net/wwlswj/article/details/106364094?`