在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