update
This commit is contained in:
@@ -9,3 +9,4 @@
|
|||||||
- tianyi07/Swin-Transformer
|
- tianyi07/Swin-Transformer
|
||||||
- qualcomm/Shufflenet-v2
|
- qualcomm/Shufflenet-v2
|
||||||
- timm/mobilenetv4_conv_small.e2400_r224_in1k
|
- timm/mobilenetv4_conv_small.e2400_r224_in1k
|
||||||
|
- nv-community/MambaVision-T-1K
|
||||||
14
enginex/MambaVision-T-1K.md
Normal file
14
enginex/MambaVision-T-1K.md
Normal file
@@ -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
|
||||||
|
```
|
||||||
Reference in New Issue
Block a user