cuda out of memory

[Richardg706]

Hi, thanks a lot for your great work! i really like your beautiful demo notebooks ,but i only have a poor RTX2080, when running the code Demo-HIPIE+SAM ,i came across the error:

RuntimeError: CUDA out of memory. Tried to allocate 60.00 MiB (GPU 0; 10.75 GiB total capacity; 9.30 GiB already allocated; 64.00 MiB free; 9.53 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

is there a simple way to solve this problem? cause i just wanna try the demo with a few pictures ,i thought this woundn't need that much of memory,thank you!!

[KKallidromitis]

Hello, sorry for the delay. I coded the HIPIE+SAM demo on google colab: colab demo
However, it is slow and a little tricky to do the installations (restarting is required) and also doesn't run on the free Tesla T4 GPU (it is too small). However, using a larger GPU you can reproduce the demo online.