用最快的方法搭建一个mamba环境:恒源云快速搭建mamba环境

部署运行你感兴趣的模型镜像

如何用最快的方法搭建一个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

回车之后没有报错,说明安装成功啦!

您可能感兴趣的与本文相关的镜像

PyTorch 2.8

PyTorch 2.8

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值