python - How to avoid "CUDA out of memory" in PyTorch - Stack Overflow

PHOTO EMBED

Thu Jul 21 2022 17:03:09 GMT+0000 (Coordinated Universal Time)

Saved by @david #python

import torch
torch.cuda.empty_cache()
content_copyCOPY

https://stackoverflow.com/questions/59129812/how-to-avoid-cuda-out-of-memory-in-pytorch/62556666#62556666