NameError: name 'selective_scan_cuda' is not defined. Did you mean: 'SelectiveScanCuda'

[hxyAhut]

No description provided.

[hxyAhut]

尝试进行分类任务的时候出现上面的问题，尝试安装mamba_ssm还是存在同样的问题，请问该如何解决呢

[MzeroMiko]

You may need to re-configurate your environment. Here's a take-away solution:

    conda create -n ${NAME} python==3.10 -y
    which conda && which python && which pip
    pip install torch==2.2 torchvision torchaudio triton pytest chardet yacs termcolor fvcore seaborn packaging ninja einops numpy==1.24.4 timm==0.4.12
    pip install tensorboard
    CUDA_HOME=/usr/local/cuda-12.1/ pip install kernels/selective_scan