【pytorch】【解决】【GPU下】RuntimeError: CUDA error: device-side assert triggered

错误:在GPU 下训练网络出现这样的错误,step=45前运行的很顺利,但是一到step=46就冒出这样的错误。一开始以为是内存不足,但是减少batch_size后并不能运行。
尝试:有博客评论讲在CPU下运行错误就明显报出来了。于是出现下述错误:
# pytorch----Target 200 is out of bounds
这就说明了有样本的标签超出了样本类别数,后来仔细想想发现,一共200个样本,标记为1200,可能是此处的错误。更改为0199后就没有问题了。
#或者可能是网络的输出个数小于样本类别数。

另外:
为什么程序总是在step=46时出现错误,毕竟样本是随即选择的阿?
因为设置了pytorch以及numpy、cudnn等的seed,每次都能保证从开始运行时随机选择的都一样,因此每次都在step=46时出现了标签为200的样本,看来设置seed还是很重要的勒!
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值