报错
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 330.00 MiB (GPU 0;2.00 GiB total capacity;1006.56 MiB already allocated;144.51 MiB free;1.02 GiB reserved in total by PyTorch) If reserved memory is>> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
做深度学习相关的实验，可以看到我的显卡内存很小（哭了，不过我有时候是在别的电脑上做的，那个电脑比这个好用），网上搜到的说的 max_split_size_mb:128 这个方法我贴到我代码上之后没有效果。
因为我在这个电脑上做的是主实验后面的一些对比实验，也就是代码中很多张量很多数据是不需要的。
所以抖了个机灵，把不需要的数据全都注释掉，代码正常运行，对比实验数据跑出来了耶！
本文转载自: https://blog.csdn.net/Lu0v0Lu/article/details/130938943
版权归原作者 YL野鹤 所有， 如有侵权，请联系我们删除。
版权归原作者 YL野鹤 所有， 如有侵权，请联系我们删除。