Linux系统CUDA10.2+CUDNN安装教程
這里主要針對(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)題。
- 上一篇: Windows系统CUDA10.2+CU
- 下一篇: Linux闲时自动抢占GPU脚本