在經過之前的各項安裝和準備工作《ubuntu從零開始安裝mxnet–安裝NVIDIA驅動》,《ubuntu從零開始安裝mxnet–安裝cuda8.0》之後。
接下來我們安裝cudnn,cudnn的安裝比較簡單,不過下載可能稍微比較麻煩些。
下載cudnn
我們先要在nvidia的官網註冊個賬號。然後才能選擇對應的cudnn版本下載安裝。根據Mxnet官網的要求我們下載5.0的版本。下載下來的內容是個壓縮包。
安裝cudnn
我們將下載下來的壓縮包解壓縮,會出來一個cuda文件夾,裏面分爲include和lib64兩個文件夾,我們需要將對應文件夾下的文件拷貝到之前安裝cuda8.0的位置下的對應文件夾下。
給個例子,
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/lib* /usr/local/cuda-8.0/lib64
測試安裝
在官網中下載cudnn_sample,解壓,然後進入
cd cudnn-sample-v5/mnistCUDNN
make
./mnistCUDNN
最後如果出現test passed表示測試成功。