Win10环境下安装Mamba和causal_conv1d

1.安装pytorch,torchvision,torchaudio,cudatookit

(1)首先我的服务器的cuda版本是12.4的,对应的python版本是3.10,。但是根据Mamba的代码,cuda只需要高于11.8就可以,所以我这里安装的pytorch,torchvision,torchaudio,cudatookit对应的都是cuda11.8版本的,安这pytorch,torchvision,torchaudio方法很多,可以:

        1.直接在官网上面https://pytorch.org/get-started/locally/按照下面的pip语句(蓝色划线处)在命令行安装即可。

        2.我是直接从清华镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/下载了对应的pytorch,torchvision,torchaudio的压缩包安装的

  对应的压缩包:

然后命令行cd到放上述压缩包的路径下,直接pip install 压缩包名称,如图

然后安装对应版本的cudatookit,也是直接在命令行pip就可以:

到这里,我的pytorch和torchvision,torchaudio,cudatookit已经安装完成。

2.安装mamba-ssm

(1)windows系统在安装mamba-ssm之前要先安装triton这个包,但是官网上的基本上都是Linux版本的,然后我在网上找到了一个大佬编译过的.whl文件,参考这个博客https://blog.csdn.net/ddrfan/article/details/130127401,下载下来的.whl文件是这个样子的:

然后命令行cd到你这个.whl文件的路径之下,使用pip install进行安装即可。后面就可以直接pip install mamba-ssm,但是我直接pip的时候报错了,所以我就把工程文件直接下载下来:

但是在mamba的工程文件的源码setup.py中我们要进行以下改动:

将:

改为:

然后cd到这个mamba这个文件的目录当中,执行以下命令:

pip install . 或者  python setup.py install

到这mamba-sam就安装上了。

3.安装causal_conv1d

windows系统不能直接使用pip语句安装causal_conv1d,于是我直接在网上下载了causal_conv1d-1.2.0.post2(网站记不住了,搜一下应该jiuneng)的压缩包,解压之后就是下图这样,

这个文件里有一个setup.py文件,同样的也需要进行源码编译:

将:

改为:

然后再cd进入到causal_conv1d-1.2.0.post2这个文件夹中,执行以下语句:

pip install . 或者  python setup.py install

 至此,我的mamba就装好了,pip list 直接可以查看

这个过程只是按照我自己的设备来的,所以不一定符合每个人的安装需求

4.网上也还有很多写的很详细的安装教程:

https://blog.csdn.net/weixin_70351416/article/details/135918564

https://blog.csdn.net/yyywxk/article/details/136071016 

 大家理性参考!!!

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值