1. 程式人生 > >ubuntu16.04更改CUDNN 6.0 版本

ubuntu16.04更改CUDNN 6.0 版本

在 ubuntu16.04 上安裝tensorflow-gpu發現沒有對應版本。檢視之後,我的電腦是cuda 8.0,cudnn 7版本

所以需要更換cudnn版本為6.0,才可以安裝tensorflow-gpu

一、下載linux版的CUDNN 6.0

官網下載:https://developer.nvidia.com/cuDNN,需要郵箱註冊。

二、安裝

1)解壓

到下載的目錄下:

tar zxvf cudnn-8.0-linux-x64-v6.0.tgz

2) 刪除原來的CUDNN檔案

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

3)cd 到之前解壓的cuda資料夾下

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

4)cd到/usr/local/cuda/lib64/資料夾下,建立軟連結

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.6.0.21
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so   
sudo ldconfig 

5)檢視版本資訊

cuda版本

cat /usr/local/cuda/version.txt

cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2