Macbook M2 Max安装xformers时出现torch模块未找到错误

问题背景 #

在M2 Max Mac的Python3.11环境中，已成功安装PyTorch 2.0.1且验证可用，但执行pip3 install xformers时，构建wheel阶段报错ModuleNotFoundError: No module named 'torch'，清理缓存、重启设备后问题仍存在。

解决方案 #

方法1：禁用构建隔离模式安装 #

pip默认会创建隔离的临时构建环境，该环境未包含已安装的torch，导致报错。使用--no-build-isolation参数让pip直接复用当前环境的依赖：

pip3 install xformers --no-build-isolation

方法2：预安装构建依赖（备选） #

如果方法1失效，可先确保构建所需依赖已安装，再执行xformers安装：

pip3 install setuptools wheel torch
pip3 install xformers

补充说明 #

M系列Mac上xformers的编译过程依赖本地已安装的PyTorch，禁用构建隔离能确保构建流程直接调用现有torch环境，从根源避免依赖缺失问题。

内容的提问来源于stack exchange，提问作者HowardC