安装 flash_attn 时 ModuleNotFoundError: No module named ‘torch‘ 错误解决

最新推荐文章于 2025-02-27 14:59:56 发布

engchina

最新推荐文章于 2025-02-27 14:59:56 发布

阅读量1w

点赞数 3

分类专栏： LINUX 文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/engchina/article/details/131522645

版权

LINUX 专栏收录该内容

851 篇文章

订阅专栏

在尝试使用pip安装flash_attn时遇到了ModuleNotFoundError:Nomodulenamedtorch的错误。这是由于系统中缺少torch库导致的。通过降级flash_attn版本到1.0.5，问题可以得到解决。参考解决方案来自于GitHub上的相关issue。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装 flash_attn 时 ModuleNotFoundError: No module named 'torch' 错误解决

安装 flash_attn 时，

pip install flash_attn

报错如下，

Collecting flash-attn (from -r requirements.txt (line 18))
  Using cached flash_attn-1.0.7.tar.gz (2.0 MB)
  Installing build dependencies ... done
  Getting requirements to build wheel ... error
  error: subprocess-exited-with-error

  × Getting requirements to build wheel did not run successfully.
  │ exit code: 1
  ╰─> [15 lines of output]
      Traceback (most recent call last):
        File "/root/miniconda3/envs/openchat/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 353, in <module>
          main()
        File "/root/miniconda3/envs/openchat/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 335, in main
          json_out['return_val'] = hook(**hook_input['kwargs'])
        File "/root/miniconda3/envs/openchat/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 118, in get_requires_for_build_wheel
          return hook(config_settings)
        File "/tmp/pip-build-env-xnmye7p3/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 341, in get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=['wheel'])
        File "/tmp/pip-build-env-xnmye7p3/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 323, in _get_build_requires
          self.run_setup()
        File "/tmp/pip-build-env-xnmye7p3/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 338, in run_setup
          exec(code, locals())
        File "<string>", line 13, in <module>
      ModuleNotFoundError: No module named 'torch'
      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
error: subprocess-exited-with-error

× Getting requirements to build wheel did not run successfully.
│ exit code: 1
╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip.

解决方法，