VGG16爆显存问题（3090）

Question

CUDA out of memory. Tried to allocate 294.00 MiB (GPU 0; 24.00 GiB total capacity; 5.12 GiB already allocated; 16.10 GiB free; 5.20 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.

以上是报错的代码，明显能看到这个显存距离用满还差着好多好多，为啥就不让我跑呢🙃

Answer 1

allocated memory try setting max_split_size_mb to avoid fragmentation 提示是 内存碎片化了。 也就是你的内存虽然很多，但是 都是碎片， 找不到一片符合指定大小的连续内存。 按照提示， 查内存碎片化 消除相关代码。