安装了 PyTorch 想使用 Metal GPU (MPS) 功能,但是竟然报错了,问了 gpt 回答
Whisper 使用的模型或其中某些层(比如位置编码或 transformer 模块)内部用了稀疏张量,而当前 PyTorch 的 MPS 后端 不支持稀疏张量操作。这不是你的代码问题,而是 PyTorch 在 macOS 上 MPS 后端支持不完整有人在 github 上提过 issue
https://github.com/pytorch/pytorch/issues/129842
所以只能用 cpu 来跑了,最后不得不说折腾完这些 mac 一格电都没掉,续航真的顶
评论 (0)