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

albertolin

已于 2024-05-15 18:00:00 修改

阅读量425

点赞数 4

文章标签： python 人工智能开发语言

于 2024-05-15 17:56:08 首次发布

本文链接：https://blog.csdn.net/albertolin/article/details/138916093

版权

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

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

操作流程如下：

conda create -n your_env_name python=3.10.14
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
pip install causal-conv1d==1.1.1
pip install mamba-ssm

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

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

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

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

关注