THUDM / CogVideo

text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

4090显卡CUDA out of memory

chenyf2022 opened this issue · comments

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 4.56 GiB (GPU 0; 23.68 GiB total capacity; 18.96 GiB already allocated; 2.84 GiB free; 18.97 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.

您好,在使用4090显卡尝试生成视频的时候出现了这个问题,请问有什么解决的方法吗?

生成视频需要25G左右的显存。我们正在修复huggingface的demo,之后可以去Huggingface上尝试生成