ubuntu從零開始安裝mxnet--安裝cudnn5.0

在經過之前的各項安裝和準備工作《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表示測試成功。
這裏寫圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章