New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

CUDA out of memory #15

Open

fallbernana123456 opened this issue Aug 5, 2024 · 1 comment

fallbernana123456 commented Aug 5, 2024

V100 在 --num_image>1 时，第一个没有问题，但是在继续生成第二个时会 CUDA out of memory，是不是第一个生成完成后没有释放资源？

Owner

catcathh commented Aug 8, 2024

可以尝试在生成一张图片后torch.cuda.empty_cache()释放资源

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment