Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM

在python上加載caffe 的網絡時會報上述錯誤,解決方案如下

方式1

1.在報錯的相應層添加 engine: CAFFE

結果:加載網絡時沒有報錯,但是net.forward前向計算時,會報Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_BAD_PARAM
解決如下

2.重新編譯caffe源碼

編譯時, USE_CUDNN := 0

結果:用cudnn編譯和不用cudnn編譯結果不一樣,因此我採用的是方式2

方式2

所有卷積層有group參數的,都加上 engine: CAFFE

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