Win10安装mamba个人总结的2种实现方式
强烈推荐大家参考这篇文章,讲述的完整且全面。
Mamba 环境安装踩坑问题汇总及解决方法_error: could not build wheels for causal-conv1d, w-CSDN博客
我的环境是win10,Pycharm2019,miniconda,Python 3.10.14,cuda 11.8和对应的cudnn。
1、DIY本地安装
操作流程如下:
- 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在本地的安装。
2、现成的docker镜像
如果嫌麻烦,可以安装Ubuntu虚拟机,借助docker pull mamba镜像到本地,请参考解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像_causal-conv1d安装失败-CSDN博客。注意修改docker为国内的源,这样速度能快一些,毕竟有4个超1G的文件和1个超4G的文件。拉取中突然卡住,一动不动只能重新拉取。多次这样的经历,会有点抓狂……
原文地址:https://blog.csdn.net/albertolin/article/details/138916093
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!