Window下环境准备

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
pip install setuptools==68.2.2
conda install nvidia/label/cuda-11.8.0::cuda-nvcc_win-64
conda install packaging
pip install triton-2.0.0-cp310-cp310-win_amd64.whl

安装方式

pip install causal_conv1d-1.1.1-cp310-cp310-win_amd64.whl
pip install mamba_ssm-1.1.3-cp310-cp310-win_amd64.whl

具体请参考本人CSDN博客及其系列博客：https://blog.csdn.net/yyywxk/article/details/140420538

注意 python torch 及 cuda 版本一定要一致。

安装有问题加CSDN博客vx