update
This commit is contained in:
11
README.md
Normal file
11
README.md
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
# enginex-hygon-tts
|
||||||
|
|
||||||
|
运行于【海光 DCU】系列算力卡的【目标检测】引擎
|
||||||
|
|
||||||
|
因具体模型之间的启动方式和具体镜像会有略微差别,请详细查看 `/enginex` 目录下各个支持模型的启动测试方式。
|
||||||
|
|
||||||
|
## 可支持模型列表
|
||||||
|
可在项目文件夹 `/enginex` 下查看具体可支持模型文件的运行方式。
|
||||||
|
|
||||||
|
支持模型列表:
|
||||||
|
- yolo_master/YOLOv5
|
||||||
10
enginex/YOLOv5.md
Normal file
10
enginex/YOLOv5.md
Normal file
@@ -0,0 +1,10 @@
|
|||||||
|
# 运行方式
|
||||||
|
|
||||||
|
```python
|
||||||
|
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
|
||||||
|
|
||||||
|
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro --name docker_name imageID bash
|
||||||
|
|
||||||
|
cd /path/workspace/
|
||||||
|
pip3 install -r requirements.txt
|
||||||
|
```
|
||||||
Reference in New Issue
Block a user