ByteDance-Seed / ByteDance-Seed/cryofm

torch.OutOfMemoryError: CUDA out of memory.

Open
#1 1 comment 0 reactions 0 assignees View on GitHub
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

Image

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.