Google Colab中flash_attn包因Subprocess-Exited-With-Error无法构建Wheel

针对你在Colab中安装flash_attn时遇到的编译失败问题，以下是经过验证的可行解决方案：

方案1：安装适配版本的预编译包 #

避开源码编译步骤，先通过!nvcc --version确认Colab当前的CUDA版本，然后安装对应适配的flash_attn版本。针对当前Colab默认的CUDA 12.x环境，可执行：

!pip install flash-attn==2.5.8 --no-build-isolation

--no-build-isolation参数会跳过编译隔离，直接使用环境中的依赖完成安装，避免编译环节的版本冲突。

方案2：降级PyTorch到兼容版本组合 #

如果预编译包安装仍有问题，可降级到经过验证的兼容版本组合：

1. 卸载现有PyTorch相关包：

!pip uninstall -y torch torchvision torchaudio

2. 安装适配CUDA 11.8的PyTorch版本：

!pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118

3. 安装兼容的flash_attn版本：

!pip install flash-attn==2.4.2

方案3：切换到旧版Python运行时 #

Colab近期更新了默认Python版本（当前为3.11），部分旧版flash_attn对Python 3.11兼容性不佳。可以切换到Python 3.10运行时：

• 点击Colab顶部菜单栏的「代码执行程序」→「更改运行时类型」
• 在「Python版本」下拉菜单中选择「Python 3.10」，保存后重新尝试安装flash_attn

问题原因说明 #

近期Colab升级了默认的PyTorch、CUDA及Python版本，flash_attn的源码编译对底层依赖的版本匹配度要求极高，编译失败通常是因为环境中的CUDA工具链、PyTorch版本与flash_attn的编译要求不匹配。使用预编译包或降级到兼容版本组合，可以绕过编译环节的冲突，快速完成安装。

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