RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

问题描述:

在将输入数据送入到网络模型的时候,报错:

RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

解决:

一开始在网上查找相关问题描述,其中绝大部分说的是显卡中缓存太多,需要清理缓存然后重新启动。
但是按照他们介绍的步骤,发现问题并没有得到解决,而且查看显卡状态,显存状态良好。
最后在 stackoverflow 上找到的解决方法

方法:

如果情况和我相同的话,我们输入的数据应该已经放在 GPU 上去了,我们可以去掉例如 .cuda() 的操作,将数据放回到 CPU 上重新运行代码,可以看到一个更加清楚地错误描述,其实根本问题是我们送了一个 Long 类型的数据(具体情况可能有差异),但是模型期望的却是一个 Float 类型的数据:
解决:

data = data.float().cuda()

具体解决可能有差异,但是思路应该是这样的。

END~

  • 6
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值