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