因為電腦漏洞的緣故，前幾天将Ubuntu18.04換成了20.04，無奈要重裝PyTorch等環境，隻有CUDA11.0以上版本才支援Ubuntu20.04，是以安裝了CUDA11.0、cuDNN8.0.5及PyTorch1.7.0。這裡記錄下過程，以備後用。
1、首先在Nvidia官網下載下傳适合自己機器的CUDA版本，官網下載下傳，本次使用 runfile 的方式安裝（用deb格式安裝的話，會在安裝過程中替換掉已經安裝好的顯示卡驅動），如圖所示：
在終端輸入： wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
下載下傳完成後輸入： sudo sh cuda_11.0.3_450.51.06_linux.run
如果電腦沒有安裝Nvidia顯示卡驅動，安裝會順利進行；若已經安裝了Nvidia驅動，會提醒移除，此時我們先Abort，如下圖，
重新輸入： sudo sh cuda_11.0.3_450.51.06_linux.run
下面界面中輸入 accept 回車，
2、根據提示的資訊，我們需要配置環境變量，打開 bashrc 檔案，
輸入： sudo gedit ~/.bashrc ，然後在檔案末尾添加下面3行（第1行是注釋），儲存
# CUDA Soft Link
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 然後重新整理環境變量，輸入： source ~/.bashrc
3、測試CUDA是否安裝成功，
輸入： nvcc -V ，顯示如下圖，說明成功，
4、安裝cuDNN 8.0.5，官網下載下傳（未注冊的話，注冊一個賬号即可），所需下載下傳包如下圖
将下載下傳的 cuDNN Library for Linux (x86_64) 解壓，複制解壓出來的檔案到安裝好的CUDA環境中，
輸入： sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
更改權限輸入： sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
接下來安裝Deb包， cuDNN Runtime Library for Ubuntu20.04(Deb)，cuDNN Developer Library for Ubuntu20.04(Deb)，cuDNN Code Samples and User Guide for Ubuntu20.04(Deb)
分别輸入：sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb
安裝結束後重新開機，測試是否安裝成功，
方法1，輸入： cp -r /usr/src/cudnn_samples_v8/ ~
cd ~/cudnn_samples_v8/mnistCUDNN/
make clean && make
./mnistCUDNN
出現Test passed！沒有報錯即安裝成功。
方法2，輸入： cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
出現 Result = PASS，說明安裝成功。
5、安裝PyTorch 1.7.0
在PyTorch官網檢視适合的版本，這裡選擇最新的穩定版本1.7.0，如下圖，
因為Ubuntu20.04預設不再安裝pip2，如果按照官網的指令使用 pip 的話會報錯，是以這裡使用 pip3 來安裝，
輸入： pip3 install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
安裝完成後進入 python3 環境，檢查PyTorch 是否安裝成功。
輸入： python3
import torch
import torchvision
print(torch.cuda.is_available())
沒有報錯，輸出True，說明安裝成功，至此結束。
補充：
Ubuntu系統安裝更新後，顯示卡驅動切換成了系統自帶的驅動，手動安裝CUDA時安裝的顯示卡驅動看不到了，
此時，可以使用單獨安裝顯示卡驅動的辦法來解決，
首先，更新apt-get 源清單
輸入：sudo apt-get update
sudo apt-get upgrade
接着，添加驅動到庫并更新
輸入： sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
然後就可以在Additional Drivers裡面找到适合的較新的驅動版本了，選擇對應的驅動，點選Apply changes，等待安裝完成後，重新開機即可。