Mamba环境安装踩坑小结

Mamba是2024年新提出的一种先进网络，其变种有MedMamba、Umamba、nnUmamba等，广泛用于分类、分割任务。然而其环境却难以成功安装，这里记录一下我在浏览各种方法之后小结的一套流程。

首先，在Linux上执行以下命令，安装虚拟环境：

conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging

接着，大多数问题出现在安装mamba依赖包的以下两条指令：

pip install causal-conv1d==1.1.1
pip install mamba-ssm

 这里不执行这两句指令，而是下载.whl文件，如下链接【免费】mamba、causal-conv1d安装.whl文件资源-CSDN文库

将该文件解压并移动到mamba工程文件夹中，cd至该文件夹下，执行如下指令：

pip install mamba_ssm-1.0.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

pip install causal_conv1d-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

安装成功即可。本人和同门均使用该步骤成功安装。