首页
Preview

sd报错RuntimeError: MPS backend out of memory

这个错误好像只有在mac 上面才会出现

RuntimeError: MPS backend out of memory (MPS allocated: 4.62 GB, other allocations: 8.06 GB, max allowed: 13.57 GB). Tried to allocate 1024.00 MB on private pool. Use PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 to disable upper limit for memory allocations (may cause system failure).

在 Mac 上,cuda 无法工作,因为它没有专用的 nvidia GPU。因此,我们必须下载特定版本的 PyTorch 才能利用 Metal Performance Shaders (MPS) 后端。Apple 的这个网页对此进行了最好的解释。

https://developer.apple.com/metal/pytorch/

安装特定版本的 PyTorch 后,应该能够简单地调用 MPS 后端。

最后再设置下环境变量

export PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0

参考:https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/9133

版权声明:本文内容由TeHub注册用户自发贡献,版权归原作者所有,TeHub社区不拥有其著作权,亦不承担相应法律责任。 如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

点赞(0)
收藏(0)
rxdue
灵台方寸山,斜月三星洞!

评论(0)

添加评论