本文主要是介绍mamba_ssm和causal-conv1d安装教程,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
一、前言
最近爆火的mamba模型,号称是可以取代transformer模型的存在,这个能不能取代,时间会告诉我们的。由于mamba模型比较新,所以在安装其环境的时候,还是有点麻烦的。mamba_ssm和causal-conv1d这两个包似乎不能在windows下安装,但是我没有尝试过(github也没有windows系统的离线包),看到其他博主这样说的,身边能有linux的还是在linux下安装吧!
由于在线安装,经常由于网络问题导致安装失败,因此本次教程使用离线安装(比较简单)。
二、开始安装
创建虚拟环境、激活虚拟环境等等此处不在赘述。
我的环境配置为:torch2.0+cu118+python3.10
因此我使用这两个安装包是
causal_conv1d-1.0.0+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
mamba_ssm-1.0.1+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
下面是这个两个安装包的github的网址,可以去下载对应版本的包
causal_conv1d:Release v1.0.0 · Dao-AILab/causal-conv1d · GitHub
mamba_ssm:Release v1.0.1 · state-spaces/mamba · GitHub
下载之后,放到对应位置,且进入到要安装的环境。(先安装causal_conv1d)
安装命令:pip install 包名
eg:pip install causal_conv1d-1.0.0+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
安装过程
然后,使用相同的方法,安装mamba_ssm。
希望以上的内容,对各位朋友,有所帮助。
这篇关于mamba_ssm和causal-conv1d安装教程的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!