这个屌丝很懒，什么也没留下！

测试时报错 RuntimeError: CUDA out of memory._训练了两个epoch后cuda out of memory

作者：从前慢现在也慢 | 2024-03-11 20:56:00

训练了两个epoch后cuda out of memory

训练神经网络模型，训练时正常，训练一个epoch后测试的时候，报错RuntimeError: CUDA out of memory. 然而我的模型并没有很大，GPU占用率也不高。


with torch.no_grad():
    (your test code)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/220595