安裝CUDA10和CUDNN760

  • 2019 年 11 月 28 日
  • 筆記

安裝CUDA10

1.去官網下載想要的cuda

https://developer.nvidia.com/cuda-toolkit-archive

2.選中一個 下載包,

按照自己電腦的情況選相應的選項,注意最後一行我選的是deb(network)方式安裝,因為如果用deb(local)的話,你可以試着選那個,然後看下面的內容,需要下載整個包,有1.7G,還是算了。這個不影響,就相當於你下載英雄聯盟是選直接下載安裝包,還是現下載騰訊遊戲安裝器這種感覺把

這裡我選擇 network, 按照如下的提示來就行了:

現download那個2.8KB的東西,在下圖右上角

然後找到那個東西的下載路徑,執行如上圖提示的語句:

sudo dpkg -i cuda-repo-ubuntu1604_10.0.130-1_amd64.deb

sudo apt-key adv –fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

此時可能會出現一個錯誤:

gpgkeys: protocol `https' not supported

解決方案:

sudo apt install gnupg-curl

sudo apt-get update

sudo apt-get install cuda

然後等待其下載就好

然後成功安裝,重啟電腦使其生效。

(註:如果你覺得通過這種方式下載較慢,那可以選擇先下載好那個1.7G的本地包那個方式安裝,依照各自情況而定,不過一般來說這個會快一些,我用了7分左右就下完了)

3.驗證我剛剛CUDA是否安裝成功

可以在 /usr/local/cuda 路徑下面找到一個samples文件夾:

進入此文件夾

cd /usr/local/cuda/samples

sudo make all -j8

完成之後,找到device_query命令,我是在 samples/1_Utilities/deviceQuery目錄 下找到的,如圖所示:

然後執行 ./deviceQuery

會顯示當前電腦上的GPU設備的詳細信息,我這是用我的筆記本電腦在作演示,所以GPU就很粗糙,

是GeForce GTX-950M 只有4046MB, 打遊戲還是夠的,

若成功顯示以上信息,至此CUDA安裝成功

安裝CUDNN

1.下載cudnn7.6.0 安裝包

https://developer.nvidia.com/cudnn

然後可能需要註冊一個帳號才能繼續,

選擇:

然後選擇自己要的版本,本文以CUDNN7.6.0為例

默認顯示的是4個最新版,我選最下面的歷史版本:Archived cuDNN Releases

然後選擇:

隨後彈出:

可以選擇下載對應的Ubantu 的三個deb包:Runtime運行環境,Developer開發庫,CodeSamples示例,然後安裝。

官網也有安裝說明:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/#installlinux-deb

我選擇直接用Linux通用版本,cuDNN Libraryfor Linux,反正照樣能用

官網也有說明:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/#installlinux-tar

2.本地操作

下載下來的文件為:

cudnn-10.0-linux-x64-v7.6.0.64.solitairetheme8

改個名後解壓縮:

mv cudnn-10.0-linux-x64-v7.6.0.64.solitairetheme8 cudnn-10.0-linux-x64-v7.6.0.64.tgz

tar -zxvf cudnn-10.0-linux-x64-v7.6.0.64.tgz

得到一個文件夾:

裏面的內容如下:

進入得到的cuda文件夾,複製一些東西到我們之前安裝cuda的路徑下(/usr/local/cuda/)

cd cuda

sudo cp lib64/lib* /usr/local/cuda/lib64/

sudo cp include/cudnn.h /usr/local/cuda/include/

接下來 進入安裝cuda的目錄:

cd /usr/local/cuda/lib64

更新cuDNN庫文件的軟連接,命令如下:

sudo chmod +r libcudnn.so.7.6.0

sudo ln -sf libcudnn.so.7.6.0 libcudnn.so.7

sudo ln -sf libcudnn.so.7 libcudnn.so

sudo ldconfig

注釋:因為我們安裝的cudnn是7.6.0,所以上面的文件後面的數字是7.6.0 或者 7

如果安裝的是其他版本的cudnn,那麼相應的數字會依據版本號而作修改

至此,CUDNN安裝完畢