CUDA out of memory. 4090. train.py

[JasonLLLLLLLLLLL]

when I run examples/train.py with a 4090(24GB). it shows cuda out memeory. May I ask what GPU do you use when run train.py? Thanks!

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB. GPU 0 has a total capacty of 23.64 GiB of which 27.44 MiB is free. Process 171543 has 77.35 MiB memory in use. Including non-PyTorch memory, this process has 22.85 GiB memory in use. Of the allocated memory 22.24 GiB is allocated by PyTorch, and 155.06 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 0%| | 0/3900 [00:00<?, ?it/s]