这个错误好像只有在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
评论(0)