Win10安装mamba个人总结的2种实现方式

强烈推荐大家参考这篇文章，讲述的完整且全面。

Mamba 环境安装踩坑问题汇总及解决方法_error: could not build wheels for causal-conv1d, w-优快云博客

我的环境是win10，Pycharm2019，miniconda，Python 3.10.14，cuda 11.8和对应的cudnn。

1、DIY本地安装

操作流程如下：

1. conda create -n your_env_name python=3.10.14
2. conda activate your_env_name
3. conda install cudatoolkit==11.8 -c nvidia
4. pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
5. conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
6. conda install packaging
7. pip install causal-conv1d==1.1.1
8. pip install mamba-ssm

其中，第4步很重要，建议从pytorch下载安装，从国内源安装貌似包大小不一样。

Triton 、causal-conv1d、mamba_ssm-1.2.0.post1建议从上面的参考文献提供的链接下载。所需积分极少，但能少走很多弯路，这里特别感谢作者“yyywxk”。

最后，利用参考文献《Mamba复现与代码解读_mamba代码-优快云博客》里的“demo推理”部分的代码，得到一样的输出，就实现了mamba在本地的安装。

2、现成的docker镜像

如果嫌麻烦，可以安装Ubuntu虚拟机，借助docker pull mamba镜像到本地，请参考解决causal_conv1d和mamba_ssm无法安装 -＞ 直接使用Mamba基础环境docker镜像_causal-conv1d安装失败-优快云博客。注意修改docker为国内的源，这样速度能快一些，毕竟有4个超1G的文件和1个超4G的文件。拉取中突然卡住，一动不动只能重新拉取。多次这样的经历，会有点抓狂……