【mamba-ssm】cuda12.4|python3.12|torch2.6.0保姆级安装手册
提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档
环境
linux
python=3.12.3
torch=2.6.0
cuda=12.4
使用pip install xxx.whl 方式安装
安装成功的关键是版本对齐
下载
需要下载两个.whl
1、causal-conv1d
causal_conv1d-1.5.0.post8+cu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl下载链接一定是1.5.0.post8
2、mamba-ssm
mamba_ssm-2.2.4+cu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl下载链接
多服务器多次测试都很顺畅
安装
pip install causal_conv1d-1.5.0.post8+cu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl
pip install mamba_ssm-2.2.4+cu12torch2.6cxx11abiFALSE-cp312-cp312-linux_x86_64.whl想必你是已经安装了很多包才看到这个帖子，安装前先清个缓存 pip cache purge
插播广告
配环境配的难受？上‘栖心-心理伙伴’，AI疗愈助手还您美丽心灵~
点击跳转 栖心-心理伙伴，小栖在这里等你呦
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
6
0- 0
已为社区贡献1条内容
所有评论(0)