VisionMamba安装
1.安装python环境
conda create -n mamba python=3.10.13 -y
conda activate mamba
2.安装torch环境
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
3.安装其他包
conda install packaging
pip install causal-conv1d==1.1.1
pip install timm==0.6.13
pip install einops
pip install transformers
pip install rope
4.注释掉selective_scan_cuda
参考Mamba 环境安装踩坑问题汇总及解决方法
报错
修改位置，直接注释
