Ubuntu上CUDA和CUDNN的安裝


Ubuntu上CUDA的安裝

  下載路徑: CUDA下載

  1. `sudo dpkg -i cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb`
  2. `sudo apt-get update`
  3. `sudo apt-get install cuda`

 

設置LD_LIBRARY_PATH:

  最簡單的方式在你的 /etc/profile 或 ~/.bashrc 文件中添加如下語句

    export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
  

  或者執行命令 sudo vi /etc/ld.so.conf.d/cuda.conf,在文件中添加如下內容:

/usr/local/cuda/lib64
/lib

  然后再執行命令:

sudo ldconfig -v

 

CUDNN

  下載網址:CUDNN (需要注冊申請,批准通過,才能下載)

  缺省的編譯不支持CUDNN,所以如果要添加CUDNN支持,要修改make/config.mk文件中的編譯選項,使之支持CUDNN。

  解壓縮下載文件,將 include 和 lib64 拷貝放置在 /usr/local/cuda 目錄下即可。我下載使用的是CUDNN V4版本

   執行如下命令修改鏈接文件:

cd /usr/local/cuda/lib64
rm -rf libcudnn.so libcudnn.so.4
ln -s libcudnn.so.4.0.4 libcudnn.so.4
ln -s libcudnn.so.4 libcudnn.so

 


免責聲明!

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



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