m0_53168353的博客

原创 RuntimeError: CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0； 6.00 GiB total capacity； 3.98

在报错的代码前后都加上这两句代码： if hasattr(torch.cuda, 'empty_cache'): torch.cuda.empty_cache()再跑就可以了原文从这里看到的记录一下我的学习过程。