14 lines
781 B
Markdown
14 lines
781 B
Markdown
# 运行方式
|
||
```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
|
||
``` |