求助！我在NWPU VHR-10遥感数据集上复现目标检测代码时，出现了显存不足的情况，已经使用了80G的A800，同时batch size设置为1.还是会出现如下的显存不足的问题，不知道是哪里出了问题
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 788.00 MiB. GPU 0 has a total capacty of 79.14 GiB of which 218.75 MiB is free. Process 210149 has 78.92 GiB memory in use. Of the allocated memory 77.63 GiB is allocated by PyTorch, and 809.10 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
求助！我在NWPU VHR-10遥感数据集上复现目标检测代码时，出现了显存不足的情况，已经使用了80G的A800，同时batch size设置为1.还是会出现如下的显存不足的问题，不知道是哪里出了问题
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 788.00 MiB. GPU 0 has a total capacty of 79.14 GiB of which 218.75 MiB is free. Process 210149 has 78.92 GiB memory in use. Of the allocated memory 77.63 GiB is allocated by PyTorch, and 809.10 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF