请选择 进入手机版 | 继续访问电脑版
查看: 267|回复: 1

pytorch runtime error: CUDNN_STATUS_MAPPING_ERROR

[复制链接]

10

主题

57

帖子

143

积分

注册会员

Rank: 2

积分
143
发表于 2018-10-15 17:10:06 | 显示全部楼层 |阅读模式
将训好的模型的参数赋值给另外一个网络,
在测试赋值是否正确的时候,在
  1. output = model(input)
复制代码
一直报错:
  1. 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上,需要增加:

  1. model = model.cuda()
复制代码
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

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