1. 程式人生 > 實用技巧 >Ubuntu16.04安裝cuda10.2 + cuDNN 7.6.5.32 (包含Nvidia驅動)

Ubuntu16.04安裝cuda10.2 + cuDNN 7.6.5.32 (包含Nvidia驅動)

此經驗帖參考了CUDA官網教程cuDNN官網安裝教程師兄經驗帖,經測試適用於惠普暗影精靈筆記本(Nvidia GTX 1660 Ti 顯示卡) & 組裝桌上型電腦(Nvidia RTX 2080 Ti 顯示卡).

1、CUDA 10.2安裝(包含Nvidia顯示卡驅動)

如果Ubuntu系統下沒有裝過Nvidia顯示卡驅動,不必單獨安裝,直接裝CUDA即可自動安裝顯示卡驅動.

(1)注意先看自己顯示卡型號是不是支援CUDA
檢視顯示卡型號的命令:
$ lspci | grep -i nvidia
支援CUDA的Nvidia顯示卡型號列表

(2)如果顯示卡支援,就可以去官網下載CUDA,選Ubuntu16.04對應的runfile(local)檔案:

CUDA 10.2官網下載地址預設下載到Home目錄下

(3)去home目錄下找到自己下載的CUDA的run檔名字,如cuda_10.2.89_440.33.01_linux.run,記下來一會要用

(4)命令列模式下安裝CUDA
先 Ctrl+Alt+F1 進入命令列模式:
$ sudo service lightdm stop
$ sudo sh ./cuda_10.2.89_440.33.01_linux.run
按照它要求的選擇accept,install,耐心等它裝完
$ sudo service lightdm start

(5)重新進入圖形介面後,配置環境檔案
$ sudo gedit ~/.bashrc
在末尾加上以下內容:

export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
  • 1
  • 2

儲存退出
$ source ~/.bashrc

檢查安裝情況:
nvidia-smi 檢查GPU,nvcc -V 檢查CUDA

更新:
嘗試用 runfile 方式在 Ubuntu 18.04下裝CUDA 10.2,安裝失敗,改用deb方式安裝成功。

方法:去官網下載deb(local)並參照官網說明安裝即可:
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal

2、cuDNN 7.6.5.32安裝

cuDNN是深度學習一般要用到的,取需

(1)官網下載
cuDNN 7.6.5.32官網下載地址
註冊後下載這三個東西,其實就是對應於ubuntu16.04的deb檔案都下載下來:
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)

(2)之後完全按照官網教程安裝和測試即可

3、跑mnist測試時可能會遇到問題

給兩臺電腦裝了cuDNN,其中一臺跑mnist測試時報錯:
bash: ./mnistCUDNN: No such file or director

我的解決辦法是把官網安裝教程裡從 2.3.2. Installing From A Debian File 到 2.4. Verifying The cuDNN Install On Linux 的所有過程重新來一遍,就解決了。