widows安装配置mamba_ssm环境

由于现在mamba大火,所以将mamba_ssm在windows中的环境配置进行介绍,如果你想在windows中进行开发,那么这是你最好的选择!!

安装步骤

1.anaconda下载

官网下载:https://www.anaconda.com/download
在这里插入图片描述
下载好后:
在这里插入图片描述
接下来点击进行安装,对于anaconda的安装,网上教程很多,请自行安装

下载CUDA

1.win+r打开cmd,输入nvidia-smi查看自己的cuda version版本。
在这里插入图片描述
2.下载cuda
下载地址:CUDA Toolkit Archive:https://developer.nvidia.com/cuda-toolkit-archive
我下载的是cuda11.8版本,因为目前我尝试的mamba_ssm环境配置中,只有cuda11.8不报错!!!!!!!!!!!!
在这里插入图片描述
下载到自己电脑上,然后点击安装,然后一路next就行
3.cudnn安装
下载地址:https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
下载下来将下面三个文件复制到cuda对应文件夹内
在这里插入图片描述

mamba_ssm下载安装

1.打开anaconda中的anaconda prompt
在这里插入图片描述
2.创建虚拟环境mamba,python版本设置为3.10.13,!!!!!别的python版本不行!!!!!!!:

conda create -n mamba python=3.10.13

3.激活创建的mamba虚拟环境:

conda activate mamba

4.下载torch和对应的torchvision和torchaudio

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

5.下载triton,这个包很重要,没有这个包,mamba_ssm包没办法下载,但是triton是专门服务于ubuntu的,没有windows版本,所以找了很久,终于找到了triton的windows环境下使用的包:
在这里插入图片描述
trion下载到本地后,需要在你下载的位置,进行安装:
在这里插入图片描述
6.安装causal_conv1d,因为causal_conv1d和mamba是一起使用的,所以需要先安装causal_conv1d,目前只有下载到本地,可以安装成功,而且这个文件是经过编译的,可以直接下载安装,网上其他方法都会报错,因为环境不匹配!!!!!!!注意
在这里插入图片描述
在这里插入图片描述

**上图是我下载到我电脑上的,编译好的,可以直接下载到本地使用,
7.安装mamba_ssm
同样的,mamba_ssm包只有下载到本地,再进行安装,才可以不报错,网上其他都有问题!!!!

下载到本地:
在这里插入图片描述
下载安装:
在这里插入图片描述

这样mamba_ssm就安装好了,可以去搞创新点和发论文楼!!!!!!!!!!!

我建了一个交流群:864452391,欢迎大家加入,群里可以探讨如何建立mamba环境,期待大家加入!!!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值