设为首页收藏本站

切换到宽版

易学智能»易学智能论坛 › 技术大讨论 › PyTorch › pytorch runtime error: CUDNN_STATUS_MAPPING_ERROR

发新帖

查看: 4554|回复: 1

pytorch runtime error: CUDNN_STATUS_MAPPING_ERROR

10 主题	57 帖子	143 积分

Rank: 2

积分: 143

发消息

发表于 2018-10-15 17:10:06 | 显示全部楼层 |阅读模式

将训好的模型的参数赋值给另外一个网络，
在测试赋值是否正确的时候，在

output = model(input)

复制代码

一直报错：

runtime error: CUDNN_STATUS_MAPPING_ERROR

复制代码

问问有没有遇到过这样的啊

回复

使用道具举报

5 主题	53 帖子	130 积分

Rank: 2

积分: 130

发消息

发表于 2018-10-15 17:15:22 | 显示全部楼层

可以参考：RuntimeError: CUDNN_STATUS_MAPPING_ERROR
应该是mode没有放在gpu上，而input在gpu上，需要增加：

model = model.cuda()

复制代码

回复

使用道具举报

发新帖

快速回复 返回顶部 返回列表