Skip to content

Instantly share code, notes, and snippets.

@djstein
Created June 30, 2023 00:19
Show Gist options
  • Save djstein/935a80155bd2e3ddd7aae77961333acb to your computer and use it in GitHub Desktop.
Save djstein/935a80155bd2e3ddd7aae77961333acb to your computer and use it in GitHub Desktop.
cuda error
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity; 5.25 GiB already allocated; 0 bytes free; 5.31 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment