2016-07-01 150 views
2

的cuDNN安裝手冊說如何將cuDNN更新爲新版本?

所有平臺

提取cuDNN文檔到你選擇的目錄,以下稱爲。 然後按照以下平臺特定的說明進行操作。

LINUX

CD 出口LD_LIBRARY_PATH =`pwd`:$ LD_LIBRARY_PATH

添加到您的構建和鏈接過程中加入-I到你的編譯 線和-L -lcudnn你的鏈接線。

看來,它只是增加了pwdLD_LIBRARY_PATH,所以我想只是替換pwd文件將做更新。

但似乎並不像後,我已經這樣做了這麼簡單,我從Theano說

cuDNN版本太舊投訴。更新到V5,是3007

回答

6
  1. dir/cuda/include/
  2. 更換cudnn.h刪除舊的庫文件dir/cuda/lib64/
  3. 添加新的庫文件dir/cuda/lib64/
+1

還要確保您的系統路徑設置正確。假設'packages /'在你的'CWD'中,是否爲 –

2

我寫了一個腳本里面用於清理安裝cuDNN版本或將現有的cuDNN更改爲更舊/更新版本。您可以從下載腳本:

https://github.com/dnzzcn/cuDNNv

這是腳本的作用:

#!/bin/bash 

rm -f /usr/include/cudnn.h 
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn* 
rm -f /usr/local/cuda-*/lib64/*libcudnn* 


cp -P packages/cudnn/include/cudnn.h /usr/include 
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/ 
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn* 

rm -rf packages/cudnn 

它您需要的版本可以自動執行安裝操作。

+0

?還需要'sudo'? – Shai

+1

是的,packages文件夾包含在版本庫目錄中。有一個python腳本調用上面的bash腳本來獲得所需的cuDNN版本;雖然我不確定,但它應該沒有sudo。 – dnzzcn

相關問題