search

czh-98 / STAR

Official code for "STAR: Skeleton-aware Text-based 4D Avatar Generation with In-Network Motion Retargeting"
https://star-avatar.github.io/
Apache License 2.0
38 stars 4 forks source link

CUDA out of memory #3

Closed XingFen-Xie closed 1 month ago

XingFen-Xie commented 2 months ago

Thank you for sharing.When I run the training code, I get the problem.CUDA out of memory.What should I do to solve this problem. Thanks.torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 0; 39.59 GiB total capacity; 37.29 GiB already allocated; 206.19 MiB free; 37.52 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

czh-98 commented 1 month ago

GPU with more than 40GB is required to train our model :).