name ‘bare_metal_version‘ is not mamba_ssm安装

目录

解决方法:

测试ok:


mamba_ssm安装报错,windows 安装时,

pip install mamba_ssm

name 'bare_metal_version' is not defined

mamba代码地址:

https://github.com/state-spaces/mamba/tree/main

解决方法:

可以试试这个博客:

Ubuntu和Windows系统之Mamba_ssm安装_mamba-ssm安装-CSDN博客

https://github.com/state-spaces/mamba/tree/main

这个安装成功:

mamba_ssm-2.2.3.post2+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64

linux pip install mamba_ssm也可以。

测试ok:

python 复制代码
import torch
from mamba_ssm import Mamba

batch, length, dim = 2, 64, 16
x = torch.randn(batch, length, dim).to("cuda")
model = Mamba(
    # This module uses roughly 3 * expand * d_model^2 parameters
    d_model=dim, # Model dimension d_model
    d_state=16,  # SSM state expansion factor
    d_conv=4,    # Local convolution width
    expand=2,    # Block expansion factor
).to("cuda")
y = model(x)
assert y.shape == x.shape

print("Success!")
相关推荐
mortimer17 分钟前
安装NVIDIA Parakeet时,我遇到的两个Pip“小插曲”
python·github
@昵称不存在38 分钟前
Flask input 和datalist结合
后端·python·flask
爱装代码的小瓶子44 分钟前
数据结构之队列(C语言)
c语言·开发语言·数据结构
赵英英俊1 小时前
Python day25
python
东林牧之1 小时前
Django+celery异步:拿来即用,可移植性高
后端·python·django
何双新2 小时前
基于Tornado的WebSocket实时聊天系统:从零到一构建与解析
python·websocket·tornado
Maybe_ch2 小时前
.NET-键控服务依赖注入
开发语言·c#·.net
超浪的晨2 小时前
Java UDP 通信详解:从基础到实战,彻底掌握无连接网络编程
java·开发语言·后端·学习·个人开发
终焉暴龙王2 小时前
CTFHub web进阶 php Bypass disable_function通关攻略
开发语言·安全·web安全·php
AntBlack2 小时前
从小不学好 ,影刀 + ddddocr 实现图片验证码认证自动化
后端·python·计算机视觉