我是靠谱客的博主 无限路人，最近开发中收集的这篇文章主要介绍AI绘画——使用stable-diffusion生成图片时提示RuntimeError: 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、结果<
最后
以上就是无限路人为你收集整理的AI绘画——使用stable-diffusion生成图片时提示RuntimeError: CUDA out of memory处理方法的全部内容，希望文章能够帮你解决AI绘画——使用stable-diffusion生成图片时提示RuntimeError: CUDA out of memory处理方法所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复