-
Notifications
You must be signed in to change notification settings - Fork 4
Open
Description
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 434.00 MiB (GPU 0; 79.15 GiB total capacity; 69.65 GiB already allocated; 267.25 MiB free; 74.75 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
When I run the traffic dataset, it occurs OOM.
Metadata
Metadata
Assignees
Labels
No labels