RuntimeError: CUDA out of memory. Tried to allocate 75.00 MiB (GPU 0; 4.00 GiB total capacity; 3.32 GiB already allocated; 0 bytes free; 38.88 MiB cached)
时间: 2023-08-14 10:47:05 浏览: 172
ArcGIS Runtime SDK for Android 100.1.0应用和开发
3星 · 编辑精心推荐
这是一个内存不足的错误,可能是由于你的程序在使用 GPU 时申请了过多的内存导致的。你可以尝试减小模型的 batch size,或者使用更大的 GPU(比如 8GB 或者 16GB 的 GPU)来解决这个问题。同时,你也可以使用 PyTorch 的内置函数来释放缓存中的内存。例如,可以使用 `torch.cuda.empty_cache()` 来释放缓存中的内存。
阅读全文