torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 52.00 MiB (GPU 0; 23.70 GiB total capacity; 343.74 MiB already allocated; 8.56 MiB free; 408.00 MiB 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
scripts/run_single_dataset.sh: line 2: 47987 Segmentation fault
Environment: GPU RTX 3090
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 52.00 MiB (GPU 0; 23.70 GiB total capacity; 343.74 MiB already allocated; 8.56 MiB free; 408.00 MiB 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 scripts/run_single_dataset.sh: line 2: 47987 Segmentation fault
Environment: GPU RTX 3090