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 的所有過程重新來一遍,就解決了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM