1. 程式人生 > >ubuntu 16.04 安裝cudnn 5.1

ubuntu 16.04 安裝cudnn 5.1

cudnn是nvidia提供的深度學習介面


一、獲取cuDNN

(需要註冊NVIDIA賬號)

選擇 cuDNN v5.1 -> cuDNN v5.1 Libraryfor Linux


二、安裝cuDNN

把標頭檔案和庫檔案拷貝到cuda的安裝目錄中,首先確保cuda的安裝目錄是/usr/local/cuda/,之前安裝cuda時,已經預設安裝目錄為/usr/local/cuda/了

sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

檔案拷貝成功



檢視cudnn版本

sudo cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2