Window Mamba 环境安装【CUDA】

Window Mamba 环境安装

  • 1. 安装 PyTorch 环境
  • 2. 直接安装 Mamba 及其依赖
  • 3. 手动编译 Mamba 及其依赖

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_conv1dmamba_ssm

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

你可能感兴趣的:(python,mamba,ssm,cuda)