pytorch遇见RuntimeError: CUDA out of memory的解决
最近在用pytorch做项目时,本人遇到RuntimeError: CUDA out of memory的错误,下面就此问题做一个记录和分享,并谈谈今后遇到爆显存问题的解决思路。目录1.查看是否其他程序占用显存2.查看pytorch和cuda是否匹配3.cuda.empty_cache()方法3.当前项目代码是否还使用了keras/tensorflow4.调小batch_size和通过程序优化显存1.查看是否其他程序占用显存遇到此类错误后,对于py格式的文件来说,程序会进行终
原创
2020-07-24 17:24:57 ·
3741 阅读 ·
0 评论