Ubuntu 17.10 安裝 NVIDIA cuDNN Library
欲使用Google Tensorflow-GPU來體驗顯示晶片運算加速深度學習、機器學習的速度,就必須要安裝NVIDIA CUDA Toolkit和cuDNN Library。本文會在Ubuntu 17.10且已安裝正確顯卡驅動程式的環境中安裝cuDNN Library!
Step 1
下載NVIDIA cuDNN
下載NVIDIA cuDNN,選擇“Library for Linux”
Step 2
安裝NVIDIA cuDNN Library
1 2 3 |
[andy@Ubuntu ~]$ tar -xvf cudnn-9.0-linux-x64-v7.1.tgz [andy@Ubuntu ~]$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ [andy@Ubuntu ~]$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ |
這邊複製的目的地是“/usr/local/cuda-9.0”的原因是之前在安裝CUDA的時候已經建立一個捷徑cuda指向cuda-9.0。
可參考安裝CUDA的第7個問題:Ubuntu 17.10 安裝 CUDA Toolkit 9.0
References