安装mamba-ssm遇CUDA版本不匹配报错，已装对应PyTorch仍未解决

错误信息 #

RuntimeError:
The detected CUDA version (11.8) mismatches the version that was used to compile
PyTorch (12.6). Please make sure to use the same CUDA versions.

完整编译日志：

ERROR: Failed building wheel for mamba-ssm
Failed to build mamba-ssm
ERROR: Failed to build installable wheels for some pyproject.toml based projects (mamba-ssm)

核心问题 #

已安装torch==2.0.1+cu118，但pip安装mamba-ssm时触发CUDA版本不匹配报错，编译失败。操作流程为：

1. 创建并激活conda环境：

   conda create -n project python=3.9
   conda activate project
   

2. 安装适配CUDA11.8的PyTorch：

   pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --extra-index-url https://download.pytorch.org/whl/cu118
   

3. 尝试安装mamba-ssm：

   pip install mamba-ssm
   

疑问解答与解决方案 #

1. pip构建是否读取了错误的PyTorch版本？ #

是。pip的**构建隔离（build isolation）**机制会在单独环境中安装依赖，默认拉取最新版PyTorch（2.7.0对应CUDA12.6），与本地CUDA11.8、torch2.0.1版本冲突。即便添加--no-build-isolation仍报错，大概率是环境残留旧缓存或依赖配置。

2. 是否需要手动编译跳过版本检查？ #

手动编译是可行方案，但优先推荐更简便的预编译/指定依赖方式。若选择手动编译，需确保CUDA环境变量指向11.8，步骤如下：

# 先安装编译依赖
pip install ninja setuptools wheel
git clone https://github.com/state-spaces/mamba.git
cd mamba
# 指定CUDA版本并编译安装
CUDA_HOME=/usr/local/cuda-11.8 pip install -e . --no-build-isolation

3. 已知可正常安装的版本组合 #

• 组合1：mamba-ssm 2.2.4 + torch 2.1.0+cu118

  # 先升级torch到适配版本
  pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
  # 安装mamba-ssm
  pip install mamba-ssm --no-build-isolation
  

• 组合2：mamba-ssm 1.2.0 + torch 2.0.1+cu118

  pip install mamba-ssm==1.2.0 --no-build-isolation
  

  注：1.1.0版本依赖torch.library.custom_op的较新实现，与torch2.0.1不兼容，故推荐1.2.0版本。

额外验证步骤 #

安装前先确认环境依赖一致性：

# 验证torch版本与CUDA匹配
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available()); print(torch.version.cuda)"
# 清理pip缓存避免残留干扰
pip cache purge

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