CUDA out of memory. Tried to allocate 38.00 MiB (GPU 0; 2.00 GiB total capacity; 1.60 GiB already allocated; 0 bytes free; 1.70 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
时间: 2023-03-25 15:03:12 浏览: 269
dc.zip_android_at1nm_triedvts_zip
这是一个关于 PyTorch 内存管理的问题,建议您参考文档中的 Memory Management 和 PYTORCH_CUDA_ALLOC_CONF 部分,尝试调整 max_split_size_mb 参数来避免内存碎片化。
阅读全文