win10跑Pytorch模型出现RuntimeError: CUDA out of memory的解决方法
2020-02-04 22:21
603 查看
Kaggle官方提供的GPU频繁断开，被迫在自己的机器上跑Kernal，显卡是贫民配置1050TI，显存4GB。训练无法开始，直接报错
RuntimeError: CUDA out of memory，尝试多种方法后解决，现在能够龟速训练了。
1.修改
Dataloader中的
num_worker参数为0（似乎仅针对windows系统下）
2.缩小
batch_size
3.把训练的代码块放入
if __name__=='__main__':中
- 点赞
- 收藏
- 分享
- 文章举报
Ztomepic
发布了34 篇原创文章 · 获赞 0 · 访问量 390
私信
关注
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 
相关文章推荐
- 出现 java.lang.OutOfMemoryError: PermGen space 错误的原因及解决方法
- 图片的cache处理时出现内存溢出(OutOfMemoryError, OutOfMemory)的解决方法
- 图片的cache处理时出现内存溢出(OutOfMemoryError, OutOfMemory)的解决方法
- SQL Server出现System.OutOfMemoryException异常的解决方法
- 高并发请求服务器时，经常出现如下异常：java.lang.OutOfMemoryError: unable to create new native thread的解决方法
- 编译android源码时，jack server 出现out of memory error 错误的解决方法
- Qt编译时出现”out of memory allocating“解决方法
- eclipse出现java.lang.OutOfMemoryError: Java heap space解决方法
- eclipse中出现java.lang.OutOfMemoryError: PermGen space解决方法
- Gradle 使用Jetty插件启动web项目时出现"java.lang.OutOfMemoryError: PermGen space"解决方法
- SQL Server出现System.OutOfMemoryException异常的解决方法
- java命令后,出现outOfMemory问题的解决方法
- 编译android源码时，jack server 出现out of memory error 错误的解决方法
- 使用 Navicat for MySQL 执行 SQL 语句时，出现 out of memory 问题的解决方法
- TomCat 出现 OutOfMemoryError : PermGen space 解决方法
- TomCat 出现 OutOfMemoryError : PermGen space 解决方法
- MyEclipse中启动出现java.lang.OutOfMemoryError: PermGen space有效解决方法
- 在Eclipse中运行Jboss时出现java.lang.OutOfMemoryError：PermGen space及其解决方法
- mysql出现out of memory错误的解决方法
- 解决Subclipse进行commit的时候出现 file out of date和remains in tree-conflict的方法