蟲匯聚之所

下午跑學習的範例程式的時候遇到了
https://github.com/tensorflow/tensorflow/issues/28326
這個提到的問題
看起來像是cuda跟cudnn版本不合

原本跑深度學習是沒什麼問題的
之前好像執行某個範例的時候炸掉,之後跑學習的時候出一堆報錯
不過一直懶得處理

今天總算提起勁
重新裝了一次 cuDNN

果不其然就能夠正常的訓練了
順便紀錄一下安裝方法

1. Unzip the cuDNN package.$ tar -xzvf cudnn-你的版本號-linux-x64-v7.tgz
2. Copy the following files into the CUDA Toolkit directory, and change the file permissions.$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

同場加映-在jupyter notebook中執行tensorboard顯示

%load_ext tensorboard
%tensorboard --logdir 資料夾位置