From 4955f20082c69a8fe0b1ca56951586e65c07233b Mon Sep 17 00:00:00 2001 From: zhousha <736730048@qq.com> Date: Thu, 4 Dec 2025 20:14:09 +0800 Subject: [PATCH] update --- README.md | 3 ++- enginex/MambaVision-T-1K.md | 14 ++++++++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 enginex/MambaVision-T-1K.md diff --git a/README.md b/README.md index 0d46a21..fb914f4 100644 --- a/README.md +++ b/README.md @@ -8,4 +8,5 @@ 支持模型列表: - tianyi07/Swin-Transformer - qualcomm/Shufflenet-v2 -- timm/mobilenetv4_conv_small.e2400_r224_in1k \ No newline at end of file +- timm/mobilenetv4_conv_small.e2400_r224_in1k +- nv-community/MambaVision-T-1K \ No newline at end of file diff --git a/enginex/MambaVision-T-1K.md b/enginex/MambaVision-T-1K.md new file mode 100644 index 0000000..d8144ab --- /dev/null +++ b/enginex/MambaVision-T-1K.md @@ -0,0 +1,14 @@ +# 运行方式 +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 +docker run -it --shm-size=1024G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name MambaVision_pytorch bash # 为以上拉取的docker的镜像ID替换,本镜像为:a4dd5be0ca23 +cd /path/your_code_data/MambaVision_pytorch/mamba +pip install wheel -i https://mirrors.aliyun.com/pypi/simple/ +pip install . --no-build-isolation --no-deps +git clone https://github.com/Dao-AILab/causal-conv1d.git +cd causal-conv1d +pip install e . +cd /path/your_code_data/MambaVision_pytorch/ +pip install . --no-build-isolation --no-deps +pip install timm==0.9.0 tensorboardX +``` \ No newline at end of file