如何用最快的方法搭建一个mamba环境
在恒源云快速搭建mamba环境
之前在本地Windows安装mamba环境,下载好各种whl文件之后,环境安装好了,但是一运行就是报错。后面还是看开了,时间才是最宝贵的,直接用恒源云的linux系统安装,尝试了一下,终于安装成功了。【恒源云选择GPU需要充money的,这个环境差不多是一块左右一小时,咱就是说,玩机器学习的,哪有不出血的,呜呜呜~】只搭建环境一两块够了,搭好测试成功之后可以关机,等到后面测试模型的时候再打开。
难点:主要在于mamba_ssm、causal_conv1d、triton版本的问题
具体步骤:
一、环境选择
1、选择PyTorch / 2.0.0 / 11.8.0 / 3.8 3090-24G环境创建实例 【注意环境要选择跟这里一样的】
2、点击下载mamba_ssm、causal_conv1d的whl文件。注意两者的下载顺序,必须是先下causal_conv1d,再下mamba_ssm。
下载之后传到恒源云上的hy-tmp/文件夹下,并在该目录下运行:
pip install causal_conv1d-1.4.0+cu118torch2.0cxx11abiFALSE-cp38-cp38-linux_x86_64.whl # 与mamba_ssm的下载顺序不能更换
pip install mamba_ssm-2.2.2+cu118torch2.0cxx11abiFALSE-cp38-cp38-linux_x86_64.whl
pip install triton==2.1.0 # 报错没关系,能跑就行
这里下载triton的时候可能会报一个错误:

但没关系,triton实际上还是下载成功了,不用管。
二、测试
运行python,进入python环境,运行:
from mamba_ssm import Mamba, Mamba2
回车之后没有报错,说明安装成功啦!
962

被折叠的 条评论
为什么被折叠?



