update
This commit is contained in:
10
README.md
10
README.md
@@ -1,3 +1,11 @@
|
||||
# enginex-hygon-fe
|
||||
|
||||
运行于【海光 DCU】系列算力卡的【特征抽取】引擎,基于 transformer 架构,支持 BGE、jina-clip 等最新流行模型
|
||||
运行于【海光 DCU】系列算力卡的【特征抽取】引擎,基于 transformer 架构,支持 BGE、jina-clip 等最新流行模型
|
||||
|
||||
因具体模型之间的启动方式和具体镜像会有略微差别,请详细查看 `/enginex` 目录下各个支持模型的启动测试方式。
|
||||
|
||||
## 可支持模型列表
|
||||
可在项目文件夹 `/enginex` 下查看具体可支持模型文件的运行方式。
|
||||
|
||||
支持模型列表:
|
||||
- iic/cv_ir50_face-recognition_arcface
|
||||
9
enginex/cv_ir50_face-recognition_arcface.md
Normal file
9
enginex/cv_ir50_face-recognition_arcface.md
Normal file
@@ -0,0 +1,9 @@
|
||||
# 运行方式
|
||||
```python
|
||||
docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10
|
||||
docker run -it --name {name} --shm-size=1024G --device=/dev/kfd --device=/dev/dri/ --privileged --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v /opt/hyhal:/opt/hyhal:ro -v {}:{} {docker_image} /bin/bash
|
||||
# 修改1 {name} 需要改为自定义名称,建议命名{框架_dtk版本_使用者姓名},如果有特殊用途可在命名框架前添加命名
|
||||
# 修改2 {docker_image} 需要需要创建容器的对应镜像名称,如: pytorch:1.10.0-centos7.6-dtk-23.04-py37-latest【镜像名称:tag名称】
|
||||
# 修改3 -v 挂载路径到容器指定路径
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
Reference in New Issue
Block a user