1.前置知識(shí)
如果Ubuntu系統(tǒng)還沒(méi)有安裝顯卡驅(qū)動(dòng)，參考這篇文章：Ubuntu20.04LTS安裝RTX-3060顯卡驅(qū)動(dòng)
2.查看顯卡驅(qū)動(dòng)版本號(hào)
當(dāng)顯卡驅(qū)動(dòng)安裝完成后，需要使用 nvidia-smi 命令查看英偉達(dá)顯卡驅(qū)動(dòng)版本。
nvidia-smi
如上圖所示，英偉達(dá)驅(qū)動(dòng)版本為520.61.05，CUDA最高支持的版本為11.8。
3.查看顯卡驅(qū)動(dòng)版本號(hào)和CUDA版本對(duì)應(yīng)關(guān)系
點(diǎn)擊該鏈接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html，查看英偉達(dá)顯卡驅(qū)動(dòng)版本和CUDA版本的對(duì)應(yīng)關(guān)系。
下圖為CUDA工具包和CUDA小版本兼容性所需的最低驅(qū)動(dòng)程序版本：
由于我工作站（Ubuntu 20.04 LTS）的英偉達(dá)驅(qū)動(dòng)版本為520.61.05，從上圖可以看出，我最高可以安裝的CUDA版本為11.8.x。（注：CUDA 12.0.x和CUDA 12.1.x都要求英偉達(dá)驅(qū)動(dòng)版本大于等于525.60.13，因此我的520.61.05不符合，所以我最高只能安裝CUDA 11.8.x的版本）。
4.查看經(jīng)典的CUDA版本號(hào)
由于我們最終是要安裝pytorch，因此選取合適的CUDA進(jìn)行安裝是是否有必要的。通過(guò)該鏈接，可以查看 CUDA與PyTorch版本對(duì)應(yīng)關(guān)系。
| PyTorch版本 | CUDA版本 |
|---|---|
| 1.2.0、1.4.0、1.5.0、1.5.1、1.6.0、1.7.0、1.7.1 | 9.2 |
| 1.0.0、1.0.1、1.1.0、1.2.0 | 10.0 |
| 1.4.0、1.5.0、1.5.1、1.6.0、1.7.0、1.7.1、1.8.1 | 10.1 |
| 1.5.0、1.5.1、1.6.0、1.7.0、1.7.1、1.8.0、1.8.1、1.8.2、1.9.0、1.9.1、1.10.0、1.10.1、1.11.0、1.12.0、1.12.1 | 10.2 |
| 1.7.0、1.7.1 | 11.0 |
| 1.8.0、1.8.1、1.8.2、1.9.0、1.9.1、1.10.0、1.10.1 | 11.1 |
| 1.8.1、1.9.0、1.9.1、1.10.0、1.10.1、1.11.0、1.12.0、1.12.1 | 11.3 |
| 1.12.0、1.12.1 | 11.6 |
從鏈接可以看出，10.2、11.1、11.3這三個(gè)版本比較經(jīng)典，能夠支持多數(shù)pytorch版本，因此可以選擇這三個(gè)版本的CUDA進(jìn)行安裝。
5.安裝CUDA
5.1.下載CUDA安裝包
CUDA官方下載鏈接：https://developer.nvidia.com/cuda-toolkit-archive
進(jìn)入CUDA官方的下載鏈接后，查找自己需要下載的版本（以CUDA 11.3.1為例）：
點(diǎn)擊自己需要下載的版本，一次選擇操作系統(tǒng)、系統(tǒng)架構(gòu)、系統(tǒng)版本和安裝方式，在這里推薦使用 runfile(local) 的安裝方式。
完成上述操作后，網(wǎng)頁(yè)下方彈出安裝的命令，如下圖所示：
5.2.執(zhí)行CUDA安裝
在Ubuntu 20.04 LTS系統(tǒng)的命令行中，按照以下命令進(jìn)行安裝。
# 第一步：使用wget命令下載安裝包
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run`
# 第二步：執(zhí)行安裝腳本
sudo sh cuda_11.3.1_465.19.01_linux.run
執(zhí)行上述命令后，等待1分鐘左右，系統(tǒng)會(huì)彈出安裝的協(xié)議，問(wèn)你 Do you accept the above EULA? 你需要在后面的光標(biāo)處，填寫(xiě) accept，然后敲回車(chē)。
然后系統(tǒng)詢問(wèn)安裝的內(nèi)容，注意！?。?一定要把Driver驅(qū)動(dòng)這個(gè)給去掉（按空格鍵可以將 X 去掉），如果[ ]內(nèi)是X 說(shuō)明是要安裝的；如果[ ] 是空，說(shuō)明不安裝。選擇完成后，然后移動(dòng)至Install 處，敲擊回車(chē)。
安裝完成后，會(huì)在 /usr/local 目錄下產(chǎn)生cuda-11.3 目錄，如下圖所示：
這樣說(shuō)明CUDA-11.3版本安裝完成了！
5.3.配置環(huán)境變量
使用 vim ~/.bashrc 命令進(jìn)行編輯，在文件末尾添加下列代碼：
# cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
然后執(zhí)行 source ~/.bashrc 刷新文件使其生效。
5.4.CUDA多版本管理
從圖中可以看出，系統(tǒng)安裝了11.1、11.3和11.6版本。
由于環(huán)境變量的地址為 /usr/local/cuda，且我們可以從圖中發(fā)現(xiàn)目錄下存在一個(gè)軟鏈接，即：/usr/local/cuda 指向了 /usr/local/cuda-11.6 鏈接，說(shuō)明此時(shí)盡管安裝了CUDA 11.1和 CUDA 11.3版本，但系統(tǒng)默認(rèn)的環(huán)境版本為11.6。如何進(jìn)行多版本的切換呢，比如想把CUDA版本切換成11.3版本（但是要保留CUDA 11.1和11.6版本），我們只需要修改軟鏈接即可，將CUDA 11.3的軟鏈接鏈接到cuda目錄下，代碼如下：文章來(lái)源：http://www.zghlxwxcb.cn/news/detail-640721.html
sudo rm -rf cuda # 刪除原有的軟鏈接
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda # 將cuda-11.3鏈接到cuda下
軟鏈接重新生成后，使用 nvcc -V 命令可以查看當(dāng)前的CUDA版本，如下所示：文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-640721.html
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Mar_21_19:15:46_PDT_2021
Cuda compilation tools, release 11.3, V11.3.58
Build cuda_11.3.r11.3/compiler.29745058_0
到了這里，關(guān)于Ubuntu20.04LTS安裝CUDA并支持多版本切換的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
