博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM
阅读量:4147 次
发布时间:2019-05-25

本文共 289 字,大约阅读时间需要 1 分钟。

在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

转载地址:http://rljti.baihongyu.com/

你可能感兴趣的文章
【Python】将txt文件转换为html
查看>>
[Linux]Shell脚本实现按照模块信息拆分文件内容
查看>>
刚刚写的一个Win32 SDK的工程扩展模板
查看>>
关于飞信的协议以及验证码
查看>>
E销助手软件(销售员的软件)
查看>>
Delphi类似QQ皮肤控件,欢迎试用!
查看>>
揭开半透明窗体的秘密(转)
查看>>
Delphi实现QQ2010的图片皮肤效果
查看>>
转载几篇别人写的皮肤类控件的技术文章
查看>>
Delphi XE的RTTI增强,动态Hook某些内部事件
查看>>
自制操作系统-kernel加载
查看>>
spring cloud config 报错,java.io.EOFException: Short read of block
查看>>
filebeat安装配置-简单版
查看>>
spring-cloud-config 突然报错 cannot open git-upload-pack
查看>>
oauth2.0 使用redis 缓存client token,导致redis内存爆了 -线上问题解决
查看>>
记录线上问题:服务器docker进程卡死了,打任何命令都无反应,最终是linux内核bug导致的
查看>>
vue devtools 安装使用(用源码构建)
查看>>
Oracle视图关联jpa失败,可以通过创建同义词解决
查看>>
Mac苹果电脑 安装virtualBox
查看>>
Nexus安装搭建和上传本地jar 详细笔记
查看>>