ubuntu cuda/cudnn 最新版安裝及檢查方法(親測有效-2021-06-25

  1. 首先nvidia-smi看一下有沒有nvidai驅(qū)動(dòng)。沒有就裝一個(gè)。

  2. 如果正常,則有驅(qū)動(dòng)。這時(shí)也會(huì)出現(xiàn)cuda版本(右上角),忽略它;

  3. nvcc -V 查看是否有cuda(nvidia-smi的cuda和nvcc的cuda不是一個(gè),見參考)

  4. 安裝cuda,在這里找到需要的cuda版本;

  5. 利用 run.sh 格式進(jìn)行安裝,這種安裝可以只安裝cuda tool,而不安裝nvidia驅(qū)動(dòng)此說明源自這里;

  6. 按照此教程安裝tool

  7. 由于已存在驅(qū)動(dòng),則會(huì)報(bào)錯(cuò),繼續(xù)按照,注意別安裝驅(qū)動(dòng)(他前面別打×)

  8. 成功后會(huì)出現(xiàn) Summary,其中Driver為未選擇

  9. 設(shè)置path路徑(summary中有安裝的路徑,按照其修改下面代碼)
    export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

  10. 安裝成功:nvcc -V 檢查

  11. 下載cudnn地址,注意要注冊賬號,注意下載的版本要對應(yīng),記得下載cuDNN Library for Linux;

  12. 注意,這里無法用wget下載,因?yàn)橐觅~號。參考此文章進(jìn)行下載,只需要把對應(yīng)的版本號改一下,版本號在步驟11處可以得到。

  13. 按照教程解壓、復(fù)制即可(注意復(fù)制到上面安裝的11.1路徑),注意新版指令發(fā)生了變化,見此評論區(qū)sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.1/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64 sudo chmod a+r /usr/local/cuda-11.1/include/cudnn*.h /usr/local/cuda-11.1/lib64/libcudnn*

  14. 利用檢測cuda/cudnn評論區(qū)方法查看安裝結(jié)果:
    cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容