(mamba_ssm)安装踩坑指南

mamba环境配置

包下载地址

https://github.com/state-spaces/mamba/releases/

https://github.com/state-spaces/mamba/releases/  mamba包下载地址
参考 https://blog.csdn.net/ggitjcg/article/details/145171758?fromshare=blogdetail&sharetype=blogdetail&sharerId=145171758&sharerefer=PC&sharesource=qq_45688164&sharefrom=from_linkimport torch
print("PyTorch Version: {}".format(torch.__version__))
import sys
print("Python Version: {}".format(sys.version))第一次安装请使用以下命令下载并安装 mamba_ssm 和 causal_conv1d(注意Python,torch和cuda版本，这里的命令是为python3.10.12,torch2.5.1+cu121环境配置的，如果你的环境不同修改对应的版号即可)：!wget https://github.com/state-spaces/mamba/releases/download/v2.2.4/mamba_ssm-2.2.4+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
!wget https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.0.post8/causal_conv1d-1.5.0.post8+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
!pip install ./mamba_ssm-2.2.4+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
!pip install ./causal_conv1d-1.5.0.post8+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
%rm ./mamba_ssm-2.2.4+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
%rm ./causal_conv1d-1.5.0.post8+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
!pip install triton

No module named 'transformers’错误

(biref)Tower:~/BiRefNet$ python mambaEnvTest.py 
Traceback (most recent call last):File "/BiRefNet/mambaEnvTest.py", line 9, in <module>from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_refFile "/anaconda3/envs/biref/lib/python3.10/site-packages/mamba_ssm/__init__.py", line 6, in <module>from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModelFile "//anaconda3/envs/biref/lib/python3.10/site-packages/mamba_ssm/models/mixer_seq_simple.py", line 20, in <module>from mamba_ssm.utils.generation import GenerationMixinFile "/anaconda3/envs/biref/lib/python3.10/site-packages/mamba_ssm/utils/generation.py", line 14, in <module>from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
ModuleNotFoundError: No module named 'transformers'
(biref) Tower:~/BiRefNet$ pip show transformers
Name: transformers
Version: 0.1
Summary: Experimental module for AST transformations.
Home-page: https://github.com/nvbn/__transformers__
Author: Vladimir Iakovlev
Author-email: [e-mail]
License: MIT
Location: /anaconda3/envs/biref/lib/python3.10/site-packages
Requires: 
Required-by: mamba-ssm
(biref) 7920-Tower:~/BiRefNet$ python
Python 3.10.16 (main, Dec 11 2024, 16:24:50) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import transformers
Traceback (most recent call last):File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'transformers'
>>> print(transformers.__version__)
Traceback (most recent call last):File "<stdin>", line 1, in <module>
NameError: name 'transformers' is not defined
>>> exit()

你的环境中安装了一个名为 transformers 的模块，但该模块似乎是一个与 Hugging Face Transformers 无关的库（版本为 0.1，描述为 "Experimental module for AST transformations"）。
你真正需要的是 Hugging Face Transformers 库，而不是这个同名的模块。

解决方法

1. 卸载冲突模块：

   pip uninstall transformers
   

2. 安装 Hugging Face Transformers：

   pip install transformers
   

3. 验证安装：
   在 Python 环境中运行以下命令，确保安装正确：

   import transformers
   print(transformers.__version__)
   

这里是引用
https://blog.csdn.net/ggitjcg/article/details/145171758