Caffe與cudnn 6.0 的兼容性問題 CUDNN_STATUS_BAD_PARAM

問題表現:訓練網絡時提示下錯誤
Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

系統環境:

  • Ubuntu 14.04
  • GeForce GTX TITAN X 驅動 384.90
  • CUDA 8.0
  • cudnn 6.0.21

由於運行 Tensorflow 1.4.0 版本時提示 libcudnn.6.so 找不到文件,因此將cudnn版本升級至6.0版本,使用 cudnn-8.0-linux-x64-v6.0.tar 解壓安裝,升級完成後,tensoflow執行正常,但caffe則提示 錯誤 CUDNN_STATUS_BAD_PARAM

問題解決:

重新安裝cudnn 5.1版本,使用 cudnn-8.0-linux-x64-v5.1.tgz 解壓安裝,覆蓋 /usr/local/cuda/lib64/ 下面的 libcudnn.so 軟鏈接文件,但 cudnn 6.0 的庫文件仍然存在

sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local

安裝完成後,caffe正常執行,tensorflow也正常執行。

其它原因:

其他博客有記錄

(1)“似乎是因爲有其他設備在使用GPU導致的,我的情況是等待一段時間就好了。”

(2)網絡配置問題,stride 設置爲1則正常

(3)顯卡內存不足


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