這里主要針對(duì)驅(qū)動(dòng)已經(jīng)安裝好了的環(huán)境，如何安裝合適的顯卡驅(qū)動(dòng)這里不多贅述，本文演示的系統(tǒng)為Ubuntu18.04，但理論上其他Linux發(fā)行版操作類(lèi)似。

驅(qū)動(dòng)適配

通過(guò)命令行輸入nvidia-smi查看自己的顯卡驅(qū)動(dòng)版本以及支持的最大CUDA版本，下圖第一行就顯示了這些信息，可以看到，最大支持CCUDA10.2，更高版本的CUDA需要升級(jí)驅(qū)動(dòng)程序。

安裝包下載

CUDA的安裝可以訪(fǎng)問(wèn)官網(wǎng)開(kāi)發(fā)者工具中找到，鏈接給出，此時(shí)顯示的結(jié)果如下，這是最新的CUDA11，點(diǎn)擊右側(cè)的紅框選擇歷史版本。

找到CUDA10.2，按照需求進(jìn)行選擇，然后安裝指示使用wget或者瀏覽器下載runfile文件。

CUDNN的下載也類(lèi)似，鏈接也給出，這個(gè)下載需要注冊(cè)開(kāi)發(fā)者賬號(hào)（簡(jiǎn)單幾步注冊(cè)登錄即可），然后如下圖選擇合適的CUDA版本對(duì)應(yīng)的CUDNN并選擇CUDNN Library for Linux。這里注意，推薦的是CUDNN8.0.2，事實(shí)上，還有一個(gè)7.6.5更合適一些，點(diǎn)開(kāi)Archived菜單查找，后者更穩(wěn)定。

安裝

CUDA安裝

下面的步驟先看完再做。

首先，通過(guò)下面的命令關(guān)閉圖形界面管理器，如果你的圖形管理器使用的是gdm3那么將命令中的lightdm換成gdm3即可。

sudo service lightdm stop

此時(shí)，桌面環(huán)境會(huì)關(guān)閉，通過(guò)快捷鍵Ctrl+Alt+F1進(jìn)入命令行界面，輸入用戶(hù)名和密碼登錄，然后找到之前下載的cuda文件（切換到該文件所在目錄下），執(zhí)行下面的命令。

sudo sh cuda_*.run

此時(shí)會(huì)進(jìn)入安裝，下面的過(guò)程不便截圖，描述一下：首先，會(huì)詢(xún)問(wèn)顯示管理器仍有開(kāi)啟，是否繼續(xù)安裝，這里選擇continue；然后，會(huì)列出一個(gè)列表要求選擇想要安裝的內(nèi)容，這里將第一個(gè)驅(qū)動(dòng)安裝的部分回車(chē)一下將那個(gè)x取消，只安裝cuda tool kit即可；之后，方向鍵選擇下面的Install進(jìn)行安裝，最后安裝成功后會(huì)有一個(gè)提示。

這時(shí)候cuda是安裝好了，但是環(huán)境變量找不到cuda程序，自己手動(dòng)修改環(huán)境變量文件，通過(guò)vim或者gedit修改，對(duì)應(yīng)命令分別為vim ~/.bashrc和gedit ~/.bashrc，將下面的內(nèi)容添加到文件最后，再通過(guò)source ~/.bashrc更新環(huán)境變量。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 export PATH=$PATH:/usr/local/cuda/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

此時(shí)，輸入nvcc -V檢查安裝是否成功，如下圖則表示安裝成功。

CUDNN安裝

切換到cudnn文件所在目錄，通過(guò)tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz命令解壓文件，會(huì)得到一個(gè)cuda文件夾，逐一執(zhí)行下面的命令進(jìn)行cudnn的安裝。

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 sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

完成后，通過(guò)下面的命令查看安裝情況，如果結(jié)果如下圖逐行顯示版本號(hào)，則安裝成功。

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

至此，在Ubuntu18.04上CUDA和CUDNN安裝完成。

訓(xùn)練測(cè)試

按照官方教程，在虛擬環(huán)境中安裝Pytorch1.5并進(jìn)行GPU訓(xùn)練測(cè)試，結(jié)果如下，成功使用CUDA訓(xùn)練。

總結(jié)

以上是生活随笔為你收集整理的Linux系统CUDA10.2+CUDNN安装教程的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。