update
This commit is contained in:
@@ -8,4 +8,7 @@
|
||||
可在项目文件夹 `/enginex` 下查看具体可支持模型文件的运行方式。
|
||||
|
||||
支持模型列表:
|
||||
- yolo_master/YOLOv5
|
||||
- yolo_master/YOLOv5
|
||||
- AI-ModelScope/YOLOv8
|
||||
- AIBS/yolov7
|
||||
- AI-ModelScope/Yolov10
|
||||
9
enginex/YOLOv8.md
Normal file
9
enginex/YOLOv8.md
Normal file
@@ -0,0 +1,9 @@
|
||||
# 运行方式
|
||||
|
||||
```python
|
||||
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-centos7.6-dtk24.04-py310
|
||||
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=80G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
|
||||
|
||||
cd /your_code_path/yolov8_pytorch
|
||||
pip install -e . -i https://mirrors.aliyun.com/pypi/simple/
|
||||
```
|
||||
9
enginex/Yolov10.md
Normal file
9
enginex/Yolov10.md
Normal file
@@ -0,0 +1,9 @@
|
||||
# 运行方式
|
||||
|
||||
```python
|
||||
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-centos7.6-dtk24.04-py310
|
||||
# 为以上拉取的docker的镜像ID替换,本镜像为:c85ed27005f2
|
||||
docker run -it --shm-size=32G -v $PWD/yolov10:/home/yolov10 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name yolov10 bash
|
||||
cd /home/yolov10
|
||||
pip install -r requirements.txt # requirements.txt
|
||||
```
|
||||
9
enginex/yolov7.md
Normal file
9
enginex/yolov7.md
Normal file
@@ -0,0 +1,9 @@
|
||||
# 运行方式
|
||||
|
||||
```python
|
||||
docker pull git.modelhub.org.cn:9443/enginex-hygon/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10
|
||||
docker run --shm-size 16g --network=host --name=yolov7_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/yolov7_migraphx:/home/yolov7_migraphx -it /bin/bash
|
||||
|
||||
# 激活dtk
|
||||
source /opt/dtk/env.sh
|
||||
```
|
||||
Reference in New Issue
Block a user