CUDA out of memory，allocate 64MB 但是2.95GB free

RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0; 6.00 GiB total capacity; 912.73 MiB already allocated; 2.95 GiB free; 944.00 MiB reserved in total by PyTorch)

显存不足，可调小batch_size
参考:

Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB_sweettea~的博客-CSDN博客 Python报错：RuntimeError: CUDA out of memory. Tried to allocate 48.00 MiB (GPU 0; 6.00 GiB total capacity; 4.44 GiB already allocated; 0 bytes free; 4.49 GiB reserved in total by PyTorch)可以修改batch_size大小进行解决，例如我是将batch_size = 128改为32即可。... https://blog.csdn.net/sweet_tea_/article/details/123263874?utm_source=app&app_version=5.5.0