RuntimeError: CUDA out of memory

阿里云双11来了!从本博客参与阿里云,服务器最低只要86元/年!

最近使用GPU训练一个用PyTorch写的神经网络模型,网络比较深,有100层左右,训练部分的时候没什么问题,然而开始测试/验证环节的时候,虽然我用了torch.cuda.empty_cache(),但是竟然直接报错:

RuntimeError: CUDA out of memory

上StackOverFlow搜了一下,搜到了相关的问题:

How to fix this strange error: “RuntimeError: CUDA error: out of memory”

解决问题的方法就是,开始测试的时候加上with torch.no_grad():;并且,在测试部分loss相加的时候使用loss.item()

https://www.jianshu.com/p/4a2be315b32a

Python量化投资网携手4326手游为资深游戏玩家推荐:《《纯白魔女》:“小伙伴”们可以更换装备啦!!

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论