Commit 90689fcf authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 1b1394b7
...@@ -32,8 +32,8 @@ CenterFace模型是一种基于单阶段人脸检测算法,作者借鉴了Cent ...@@ -32,8 +32,8 @@ CenterFace模型是一种基于单阶段人脸检测算法,作者借鉴了Cent
-v 路径、docker_name和imageID根据实际情况修改 -v 路径、docker_name和imageID根据实际情况修改
```bash ```bash
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10.1-py38 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=32G -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
cd /your_code_path/centerface_pytorch/ cd /your_code_path/centerface_pytorch/
pip3 install -r requirements.txt pip3 install -r requirements.txt
...@@ -46,7 +46,7 @@ pip3 install -r requirements.txt ...@@ -46,7 +46,7 @@ pip3 install -r requirements.txt
```bash ```bash
cd ./docker cd ./docker
docker build --no-cache -t centerface:latest . docker build --no-cache -t centerface:latest .
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=32G -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
cd /your_code_path/centerface_pytorch/ cd /your_code_path/centerface_pytorch/
pip3 install -r requirements.txt pip3 install -r requirements.txt
...@@ -57,8 +57,8 @@ pip3 install -r requirements.txt ...@@ -57,8 +57,8 @@ pip3 install -r requirements.txt
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/ 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
``` ```
DTK软件栈:dtk23.10.1 DTK软件栈:dtk24.04.1
python:python3.8 python:python3.10
torch:2.1.0 torch:2.1.0
torchvision:0.16.0 torchvision:0.16.0
``` ```
......
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