How much CUDA memory is needed for infer.py?

[KaqiKaqi]

When I tried python infer.py --config config.yaml data.image_list=[example_images/a_pikachu_with_smily_face.webp,], I got RuntimeError: CUDA out of memory. Tried to allocate 288.00 MiB (GPU 0; 9.78 GiB total capacity; 6.76 GiB already allocated; 100.56 MiB free; 6.94 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.