an illegal memory access was encountered问题解决方法之一

作为一个刚入门深度学习的新手,我在把一个深度学习模型嵌入到另一个大的框架里面时,遇到了这个问题,耗时一天半才解决这个问题

我遇到这个问题是把数据和模型送入gpu和cpu的问题,把原本送入cpu的数据在gpu中计算,才导致上面出现的内存非法占用。我的本意是把模型送入gpu中计算,代码是这样写的

self.modal.device = torch.device('cuda')

但是调试发现,模型输出的数据device显示是在cpu上,所以造成是内存非法占用,然后我把模型送到GPU的语句改成以下语句,解决了错误

self.modal = self.modal.to('cuda')

以上是我解决报错的过程,写的不太专业,各位大佬批评指正。

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值