首页 > 其他 > 详细

利用GPU训练网络时遇到的一些问题

时间:2019-09-10 00:45:30      阅读:124      评论:0      收藏:0      [点我收藏+]

1. OSError: [Error 12] Cannot allocate memory

解决办法: 出现这个错误时我是绝望的。。。因为我看了别人的解决办法,要加内存条才能解决。。。但是我不甘心,想了很久——发现了一个解决办法:将 DataLoader 中的 num_workers设置为1。缺点:数据加载必然变得很慢,可怜我的内存条只有 8 个G

2. RuntimeError: CUDA error: device-side assert triggered

解决办法: 检查类别数目设置是否有误!标签分类应该从 0 开始,如果共有10类,则类别应是从 0 开始直到 19

3. CUDA error: out of memory

解决办法: 可以适当缩小 batch size 来解决

利用GPU训练网络时遇到的一些问题

原文:https://www.cnblogs.com/xxxxxxxxx/p/11494734.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!