安装flash_attn==2.7.4.post1失败。torch环境已安装

Author: Jonathan-WeiCreated Dec 24, 2025Updated Jun 8, 2026

执行到安装flash_attn时报错。torch环境已安装,哪位大佬帮忙看下。万分感谢

(multitalk) feixin@feixin:/data/deploy/InfiniteTalk$ pip install flash_attn==2.7.4.post1
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting flash_attn==2.7.4.post1
  Using cached https://pypi.tuna.tsinghua.edu.cn/packages/11/34/9bf60e736ed7bbe15055ac2dab48ec67d9dbd088d2b4ae318fd77190ab4e/flash_attn-2.7.4.post1.tar.gz (6.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
  ╰─> [17 lines of output]
      Traceback (most recent call last):
        File "/opt/miniconda/envs/multitalk/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 389, in <module>
          main()
        File "/opt/miniconda/envs/multitalk/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 373, in main
          json_out["return_val"] = hook(**hook_input["kwargs"])
        File "/opt/miniconda/envs/multitalk/lib/python3.10/site-packages/pip/_vendor/pyproject_hooks/_in_process/_in_process.py", line 143, in get_requires_for_build_wheel
          return hook(config_settings)
        File "/tmp/pip-build-env-eqhjr_f7/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 331, in get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=[])
        File "/tmp/pip-build-env-eqhjr_f7/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires
          self.run_setup()
        File "/tmp/pip-build-env-eqhjr_f7/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 512, in run_setup
          super().run_setup(setup_script=setup_script)
        File "/tmp/pip-build-env-eqhjr_f7/overlay/lib/python3.10/site-packages/setuptools/build_meta.py", line 317, in run_setup
          exec(code, locals())
        File "<string>", line 22, 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: Failed to build 'flash_attn' when getting requirements to build wheel
(multitalk) feixin@feixin:/data/deploy/InfiniteTalk$ pip list | grep torch
torch                      2.4.1+cu121
torchaudio                 2.4.1+cu121
torchvision                0.19.1+cu121