【发布时间】:2022-11-11 03:20:24
【问题描述】:
我按照https://www.howtogeek.com/830179/how-to-run-stable-diffusion-on-your-pc-to-generate-ai-images/#autotoc_anchor_2 中描述的说明安装了 Stable Diffusion v1.4
我的机器大大超过了运行稳定扩散的最低要求: 视窗 11 专业版 第 11 代 Intel i7 @ 2.30GHz 最新的 NVIDIA GeForce GPU 16GB内存 1TB固态硬盘
但是,尝试运行测试提示时出现错误 python scripts/txt2img.py --prompt "pablo picasso 的一只猫的特写肖像,生动,抽象,色彩丰富,充满活力" --plms --n_iter 5 --n_samples 1
运行时错误:CUDA 内存不足。尝试分配 1024.00 MiB(GPU 0;8.00 GiB 总容量;6.13 GiB 已分配;0 字节空闲;PyTorch 总共保留 6.73 GiB)如果保留的内存是 >> 分配的内存,请尝试设置 max_split_size_mb 以避免碎片。请参阅内存管理和 PYTORCH_CUDA_ALLOC_CONF 的文档
阅读 Marco Ramos 的一篇文章,它似乎与 PyTorch 中的工人数量有关 Strange Cuda out of Memory behavior in Pytorch
如何在运行稳定扩散时更改工作人员的数量?如果我的机器还有很多内存,为什么会抛出这个错误?有没有人在运行稳定扩散时遇到过同样的问题?
【问题讨论】:
标签: python windows pytorch command-line-interface