Commit 0d5d0b15 authored by renzhc's avatar renzhc
Browse files

update readme

parent 5a1b5a92
...@@ -14,7 +14,7 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dt ...@@ -14,7 +14,7 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dt
```shell ```shell
# <your IMAGE ID or NAME>用以上拉取的docker的镜像ID或名称替换 # <your IMAGE ID or NAME>用以上拉取的docker的镜像ID或名称替换
docker run -it --name=mobilenetv2 --network=host --ipc=host --shm-size=16g --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/mobilenetv2_mmcv:/home/mobilenetv2_mmcv <your IMAGE ID> bash docker run -it --name=mobilenetv2 --network=host --ipc=host --shm-size=16g --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/yhal:/opt/hyhal:ro -v $PWD/mobilenetv2_mmcv:/home/mobilenetv2_mmcv <your IMAGE ID> bash
``` ```
克隆git仓库,并安装相关依赖 克隆git仓库,并安装相关依赖
...@@ -28,7 +28,7 @@ pip install -r requirements.txt ...@@ -28,7 +28,7 @@ pip install -r requirements.txt
### Dockerfile(方法二) ### Dockerfile(方法二)
cd mmpretrain/docker cd mmpretrain-mmcv/docker
docker build --no-cache -t mmpretrain:latest . docker build --no-cache -t mmpretrain:latest .
docker run -it --name=mmpretrain --network=host --ipc=host --shm-size=16g --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/mmpretrain:/home/mmpretrain <your IMAGE ID> bash docker run -it --name=mmpretrain --network=host --ipc=host --shm-size=16g --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/mmpretrain:/home/mmpretrain <your IMAGE ID> bash
pip install -e . pip install -e .
...@@ -66,9 +66,7 @@ pip install -r requirements.txt ...@@ -66,9 +66,7 @@ pip install -r requirements.txt
## 示例 ## 示例
本仓库中提供了几个在tiny imagenet下进行测试的脚本 本仓库中提供了几个在tiny-imagenet下进行测试的脚本,用8卡从零开始训练resnet50的运行方式如下,
如用8卡从零开始训练resnet50的运行方式如下,
```shell ```shell
bash tools/dist_train.sh resnet50-test.py 8 bash tools/dist_train.sh resnet50-test.py 8
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment