一绝编、CUDA 10.1安裝
??直接運行cuda_10.1.105_418.39_linux.run
,然后按照要求一步步操作就可以了。這里需要注意的是棋蚌,因為已經(jīng)安裝了其他版本的cuda和驅(qū)動雅倒。所以璃诀,在安裝的時候,不需要安裝驅(qū)動蔑匣。
??通過空格鍵取消Driver選項就可以了劣欢,最后通過回車鍵選中Install安裝。
二裁良、cuDNN v8.0.5安裝
??其實凿将,這里最主要想說的是cudnn的安裝,安裝的過程碰到了一點問題价脾。剛開始牧抵,按照以前的方法,直接執(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ù)制就可以了谎脯。
??還有一點需要注意葱跋,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
飞苇。