本文主要是介绍Vision Mamba 环境安装 运行笔记，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
环境描述
- ubuntu
- pytorch 2.1 和配套的torchvision torchaudio
- cuda 12.1
- python 3.10
安装causal-conv1d
参考复现U-Mamba。下载whl包，安装causal_conv1d==1.1.0。具体而言，
git clone http://github.com/Dao-AILab/causal-conv1d.git
git checkout v1.1.0
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
安装mamba-ssm
在Vim目录下，修改mamba-1p1p1/setup.py，避免其自动安装causal_conv1d。
python_requires=">=3.7",install_requires=["torch","packaging","ninja","einops","triton","transformers",# "causal_conv1d==1.1.0",],
参考ca
这篇关于Vision Mamba 环境安装 运行笔记的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
