Window Mamba 环境安装【CUDA】

1. 安装 PyTorch 环境

conda create -n mamba python=3.10
conda activate mamba
conda install cudatoolkit==11.8
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
# path to triton
pip install ./triton-2.0.0-cp310-cp310-win_amd64.whl

下载地址：triton

2. 直接安装 Mamba 及其依赖

# path to causal_conv1d
pip install ./causal_conv1d-1.1.1-cp310-cp310-win_amd64.whl
# path to mamba_ssm
pip install ./mamba_ssm-1.1.3-cp310-cp310-win_amd64.whl

下载地址：causal_conv1d，mamba_ssm

3. 手动编译 Mamba 及其依赖

3 和 2 二选一即可

pip install setuptools==68.2.2
conda install nvidia/label/cuda-11.8.0::cuda-nvcc_win-64
conda install packaging

然后按照 [1] 编译安装 causal_conv1d 和 mamba_ssm。

[1]: Window 下Mamba 环境安装踩坑问题汇总及解决方法 （无需绕过selective_scan_cuda）