ubuntu 16.04 下安裝NVDIA的庫(CUDA+CUDNN)


  立志學習神經網絡的同學,為了讓它能夠跑快一點,估計英偉達的GPU是要折騰一番的.

首先看一下什么是CUDA ?

  CUDA(Compute Unified Device Architecture),是顯卡廠商NVIDIA推出的運算平台。 CUDA™是一種由NVIDIA推出的通用並行計算架構,該架構使GPU能夠解決復雜的計算問題。

它是一個標准API的並行計算平台.它允許軟件開發人員和工程師使用CUDA提供的GPU計算單元.

  什么是CUDNN?

NVIDIA cuDNN是用於深度神經網絡的GPU加速庫。它強調性能、易用性和低內存開銷。NVIDIA cuDNN可以集成到更高級別的機器學習框架中,如加州大學伯克利分校的流行caffe軟件。簡單的,插入式設計可以讓開發人員專注於設計和實現神經網絡模型,而不是調整性能,同時還可以在GPU上實現高性能現代並行計算.

怎么安裝CUDA呢?

這個見我以前的博文.

怎么安裝CUDNN?

1 首先去下載:

https://developer.nvidia.com/rdp/cudnn-archive

2 安裝deb文件:解壓縮庫文件並把頭文件考入庫中:

tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz   sudo cp include/cudnn.h /usr/local/cuda/include/   sudo cp lib64/lib* /usr/local/cuda/lib64/

3 安裝deb文件:

  

sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb 

sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb

sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb

 

4 驗證是否成功:

  

cd /usr/src/cudnn_samples_v6/mnistCUDNN
sudo make clean && make
sudo ./mnistCUDNN

  log信息:

Result of classification: 1 3 5

Test passed!

說明測試安裝成功.

 

參考文檔:

https://blog.csdn.net/qq_35239859/article/details/79827203


免責聲明!

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



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