RuntimeError: CUDA out of memory about raft

[hajoxs]

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 59.41 GiB (GPU 0; 6.00 GiB total capacity; 636.69 MiB already allocated; 2.60 GiB free; 1.37 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

I'm also experiencing this problem and would like to know if you've resolved it yet

[ismeyueyue]

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 59.41 GiB (GPU 0; 6.00 GiB total capacity; 636.69 MiB already allocated; 2.60 GiB free; 1.37 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

I'm also experiencing this problem and would like to know if you've resolved it yet

Yes, I have soved this proble with resize the input image to low resolution, like 720p, 480p. Here is my code, you can refer:

def resize_frame(frame, target_height=480):
    """
    Resize a frame while maintaining aspect ratio to make its height close to the target height.
    """
    height, width, _ = frame.shape
    scale_percent = target_height / height

    new_width = int(width * scale_percent)
    resized_frame = cv2.resize(frame, (new_width, target_height), interpolation=cv2.INTER_AREA)
    return resized_frame

[hajoxs]

Thank you for your help