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 环境安装踩坑问题汇总及解决方法
报错
修改位置，直接注释