Vision Mamba 环境安装 运行笔记

本文主要是介绍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 环境安装 运行笔记的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---