PyTorch出现”RuntimeError: CUDA out of memory“的解决方法_outofmemoryerror: cuda out of memory. tried to all

作者：从前慢现在也慢 | 2024-02-16 00:42:35

踩

outofmemoryerror: cuda out of memory. tried to allocate 7.12 gib (gpu 0; 24.

问题描述

在运行以下代码时遇到了内存不足的问题。


# 代码
x = self.conv(x)
 
# 出现以下问题
RuntimeError: CUDA out of memory. Tried to allocate 66.00 MiB (GPU 0; 2.00 GiB total capacity; 1.21 GiB already allocated; 19.68 MiB free; 1.23 GiB reserved in total by PyTorch)

解决方法

参考链接：https://blog.csdn.net/pursuit_zhangyu/article/details/88717635、no_grad

将以上代码进行修改，禁止梯度计算来减少内存消耗


with torch.no_grad():
    x = self.conv(x)

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