Ubuntu 16.04 上安裝 CUDA 10.1和cuDNN v8.0.5

一、CUDA 10.1安裝

??直接運(yùn)行cuda_10.1.105_418.39_linux.run,然后按照要求一步步操作就可以了。這里需要注意的是,因?yàn)橐呀?jīng)安裝了其他版本的cuda和驅(qū)動。所以,在安裝的時候,不需要安裝驅(qū)動。


??通過空格鍵取消Driver選項(xiàng)就可以了,最后通過回車鍵選中Install安裝。

二、cuDNN v8.0.5安裝

??其實(shí),這里最主要想說的是cudnn的安裝,安裝的過程碰到了一點(diǎn)問題。剛開始,按照以前的方法,直接執(zhí)行命令:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

??看命令的第一行,直接把cudnn.h復(fù)制到cuda-10.1的安裝目錄下。問題就出在這里,復(fù)制之后執(zhí)行命令cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看cudnn版本,結(jié)果發(fā)現(xiàn)沒任何輸出。
??找到cudnn.h這個文件,發(fā)現(xiàn)文件的內(nèi)容和較早cuDNN v7.0.5版本少了很多內(nèi)容。于是,找到官方安裝文檔https://docs.nvidia.com/deeplearning/cudnn/archives/cudnn-803/install-guide/index.html。官方文檔給出的命令為:

$ 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*

??注意到,第一行命令和以前不一樣,然后進(jìn)到文件夾仔細(xì)查看這些cudnn*.h這些文件。發(fā)現(xiàn)官方只是把以前的cudnn.h的內(nèi)容拆分到了多個文件中。這里只需要按照官方的指令重新復(fù)制就可以了。
??還有一點(diǎn)需要注意,cudnn的版本信息被放到了cudnn_version.h文件中,所以,這里想要查看cudnn的版本信息,應(yīng)該使用指令cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2。

??PS:安裝軟件還是要先看官方文檔,以官方文檔為準(zhǔn),避免安裝的時候出現(xiàn)問題。

三、多版本cuda切換

??可以看到這里有3個版本的cuda,我們只需要在/usr/local目錄下新建一個軟鏈接cuda,直接讓cuda指向某一個版本的cuda文件夾就可以了。比如,這里我們使用最新的10.1版本:sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda。以后,如果想切換其他版本的cuda,只需要修改軟鏈接的指向。
??最后,將/usr/local/cuda寫入到環(huán)境變量。通過vi ~/.bashrc打開文件,然后將以下命令加入到文件末尾:

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

然后source一下,使其立即生效:source ~/.bashrc

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

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

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