ByteDance-Seed / ByteDance-Seed/cryofm
torch.OutOfMemoryError: CUDA out of memory.
Open
- Dominant language
- Python
- Stars
- 36
- Forks
- 6
- PR merge metrics
- No merged PRs in 30d
Description
Dear cryofm team,
I encountered the issue shown in the figure when using the **cfm denoise** command. I am working with 4×16GB GPUs with 4 multiprocess, but the program still reports a CUDA out-of-memory error.
To troubleshoot this, I have already tried setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to reduce memory fragmentation as recommended by the output log but failed
.
Could you please provide a feasible solution or official low-memory configuration recommendations for running cfm denoise on 16GB GPUs?
Best,
Verdandy
Contributor guide
Assessment
This issue has not been assessed yet.