测试环境：
anaconda3 + python3.10
pip list
Package Version------------------ ------------attrs 24.3.0Automat 24.8.1buildtools 1.0.6causal-conv1d 1.1.1certifi 2024.12.14cffi 1.15.0charset-normalizer 3.4.1colorama 0.4.6constantly 23.10.4docopt 0.6.2einops 0.8.0filelock 3.16.1fsspec 2024.12.0furl 2.1.3greenlet 3.1.1huggingface-hub 0.27.0hyperlink 21.0.0idna 3.10incremental 24.7.2Jinja2 3.1.5mamba_ssm 1.1.3MarkupSafe 3.0.2mpmath 1.3.0networkx 3.4.2ninja 1.11.1.3numpy 1.24.1orderedmultidict 1.0.1packaging 24.2pillow 11.0.0pip 24.2pycparser 2.22python-dateutil 2.9.0.post0PyYAML 6.0.2redo 3.0.0regex 2024.11.6requests 2.32.3safetensors 0.4.5setuptools 68.2.2simplejson 3.19.3six 1.17.0SQLAlchemy 2.0.36sympy 1.13.3tokenizers 0.21.0tomli 2.2.1torch 2.1.1+cu118torchaudio 2.1.1+cu118torchvision 0.16.1+cu118tqdm 4.67.1transformers 4.47.1triton 2.1.0Twisted 24.11.0typing_extensions 4.12.2urllib3 2.3.0wheel 0.44.0zope.interface 7.2
测试代码：
import torchfrom mamba_ssm import Mambabatch, length, dim = 2, 64, 16x = torch.randn(batch, length, dim).to("cuda")
model = Mamba(d_model=dim,d_state=16,d_conv=4,expand=2).to("cuda")
y = model(x)assert y.shape == x.shapeprint('success')
运行结果：
Topaz Video AI
一款工业级别的视频增强软件
388 查看详情
立即学习“Python免费学习笔记（深入）”；
![[python]windows安装mamba后测试代码](https://cdn.chuangxiangniao.com/www/2025/11/175832761373541.jpg)
以上就是[python]windows安装mamba后测试代码的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/753272.html
微信扫一扫 
支付宝扫一扫 
