ModuleNotFoundError：在 Macbook M2 Max 上安装 xformers 进行稳定扩散时，没有名为“torch”的模块

Question

我正在研究稳定扩散并尝试安装 xformers 来训练我的 Lora。

\n

我的电脑是Macbook M2 Max并且已经安装了最新的python3.11和pip 23.1.2

\n

which python3\n/Library/Frameworks/Python.framework/Versions/Current/bin/python3\npython3 -V                                                                             \nPython 3.11.3\n\nwhich pip3\n/Library/Frameworks/Python.framework/Versions/Current/bin/pip3\npip3 -V\npip 23.1.2 from /Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/pip (python 3.11)\n

\n

但每次我使用下面的cmd时：

\n

pip3 install xformers\n

\n

我会收到如下消息：

\n

Collecting xformers\n  Using cached xformers-0.0.20.tar.gz (7.6 MB)\n  Installing build dependencies ... done\n  Getting requirements to build wheel ... error\n  error: subprocess-exited-with-error\n  \n  \xc3\x97 Getting requirements to build wheel did not run successfully.\n  \xe2\x94\x82 exit code: 1\n  \xe2\x95\xb0\xe2\x94\x80> [21 lines of output]\n      Traceback (most recent call last):\n        File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 353, in <module>\n          main()\n        File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 335, in main\n          json_out['return_val'] = hook(**hook_input['kwargs'])\n                                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n        File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 118, in get_requires_for_build_wheel\n          return hook(config_settings)\n                 ^^^^^^^^^^^^^^^^^^^^^\n        File "/private/var/folders/05/3wghcmr93436gfgb3ty5mnmm0000gn/T/pip-build-env-3da38ps7/overlay/lib/python3.11/site-packages/setuptools/build_meta.py", line 341, in get_requires_for_build_wheel\n          return self._get_build_requires(config_settings, requirements=['wheel'])\n                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n        File "/private/var/folders/05/3wghcmr93436gfgb3ty5mnmm0000gn/T/pip-build-env-3da38ps7/overlay/lib/python3.11/site-packages/setuptools/build_meta.py", line 323, in _get_build_requires\n          self.run_setup()\n        File "/private/var/folders/05/3wghcmr93436gfgb3ty5mnmm0000gn/T/pip-build-env-3da38ps7/overlay/lib/python3.11/site-packages/setuptools/build_meta.py", line 488, in run_setup\n          self).run_setup(setup_script=setup_script)\n                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^\n        File "/private/var/folders/05/3wghcmr93436gfgb3ty5mnmm0000gn/T/pip-build-env-3da38ps7/overlay/lib/python3.11/site-packages/setuptools/build_meta.py", line 338, in run_setup\n          exec(code, locals())\n        File "<string>", line 23, in <module>\n      ModuleNotFoundError: No module named 'torch'\n      [end of output]\n  \n  note: This error originates from a subprocess, and is likely not a problem with pip.\nerror: subprocess-exited-with-error\n\n\xc3\x97 Getting requirements to build wheel did not run successfully.\n\xe2\x94\x82 exit code: 1\n\xe2\x95\xb0\xe2\x94\x80> See above for output.\n\nnote: This error originates from a subprocess, and is likely not a problem with pip.\n

\n

我已经使用官方网站的命令安装了 PyTorch：

\n

pip3 install torch torchvision torchaudio\n

\n

当我使用以下代码运行 test.py 时：

\n

import torch\nx = torch.rand(5, 3)\nprint(x)\n

\n

我会得到正确的结果，如下所示：

\n

python3 a.py\ntensor([[0.8204, 0.4140, 0.5262],\n        [0.1857, 0.3604, 0.3496],\n        [0.0620, 0.2303, 0.1033],\n        [0.8081, 0.2826, 0.2922],\n        [0.1685, 0.5701, 0.9935]])\n

\n

这表明割炬已正确安装。

\n

我还可以使用 pip3 list 看到 torch

\n

torch              2.0.1\ntorchaudio         2.0.2\ntorchvision        0.15.2\n

\n

我还使用清除了所有 pip3 缓存

\n

pip cache purge\n

\n

这不起作用。

\n

我也重启电脑，还是不行。

\n

我看到很多问答都在谈论conda，我还没有安装conda所以我相信这不应该是conda相关的问题。

\n

如何解决这个问题？谢谢~

\n