当前位置:   article > 正文

Win10环境下安装Mamba和causal_conv1d_causal-conv1d==1.2.0.post2 源码下载

causal-conv1d==1.2.0.post2 源码下载

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 

 大家理性参考!!!

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号