AI绘画——使用stable-diffusion生成图片时提示RuntimeError: CUDA out of memory处理方法_stable diffusion cuda out of memory

作者：AllinToyou | 2024-02-18 04:30:13

踩

stable diffusion cuda out of memory

提示显存不足

RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB (GPU 0; 8.00 GiB total capacity; 5.62 GiB already allocated; 109.75 MiB free; 5.74 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

1、添加参数–n_samples 1，其默认值为3

python scripts/txt2img.py --prompt “a photograph of an astronaut riding a horse” --plms –n_samples 1

在这里插入图片描述

2、降低精度

修改scripts/txt2img.py文件，在model.to(device)后添加.half()
在这里插入图片描述

3、结果<

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/104725