1. 程式人生 > 實用技巧 >Centos cudnn版本升級

Centos cudnn版本升級

@

用Centos伺服器執行程式碼時,程式碼報錯CuDnn版本不夠,我的版本是7.5.1,但是程式碼的要求是基於7.6.0的CuDnn版本,所以就解除安裝了舊版本,安裝了7.6.0的CuDnn。我的Cuda是10.0.130,升級完以後tensorflow可以正常使用。

下載新版本

cudnn下載連結:
https://developer.nvidia.com/rdp/cudnn-archive

選定版本點進去有針對不同系統的軟體包,最開始我是下載的針對Centos的rpm包,但是沒有安裝成功,個人感覺這個有點麻煩,還是下.tgz包方便一點。下載需要註冊一個nvidia帳號,最後通過郵箱啟用,很快。那個cuDNN Library for Linux就是.tgz包的下載連結

在這裡插入圖片描述
你會得到一個類似這樣的檔案,你可以手動把字尾改成.tgz,改成cudnn-10.0-linux-x64-v7.6.0.64.tgz,然後用如下命令解壓
tar -zxvf cudnn-10.0-linux-x64-v7.6.0.64.tgz
解壓後是一個名為cuda的資料夾,資料夾中有include和lib64兩個資料夾。

刪除舊版本

在這裡插入圖片描述
我的資料夾是這樣的,注意到cuda-10.0的include資料夾也有cudnn.h,所以我把這兩個資料夾都執行了以下的操作。
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

sudo rm -rf /usr/local/cuda-10.0/include/cudnn.h
sudo rm -rf /usr/local/cuda-10.0/lib64/libcudnn*

安裝新版本

cd進剛解壓的cuda資料夾
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

建立軟連線

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.6.0
sudo ln -sf libcudnn.so.7.6.0 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so

sudo ldconfig

測試檢視cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
在這裡插入圖片描述
升級成功