本文共 289 字,大约阅读时间需要 1 分钟。
在python上加载caffe 的网络时会报上述错误,解决方案如下
结果:加载网络时没有报错,但是net.forward前向计算时,会报Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_BAD_PARAM
解决如下编译时, USE_CUDNN := 0
结果:用cudnn编译和不用cudnn编译结果不一样,因此我采用的是方式2
所有卷积层有group参数的,都加上 engine: CAFFE
转载地址:http://rljti.baihongyu.com/