34 lines
1.0 KiB
Markdown
34 lines
1.0 KiB
Markdown
# enginex-ascend-910-vc-cnn
|
||
|
||
运行于【天数智芯-智铠】系列算力卡的【视觉分类】引擎,基于 CNN 架构,支持 BEiT、MobileViT 等流行模型
|
||
|
||
## QuickStart
|
||
|
||
1、从 modelscope上下载视觉分类的模型,例如 microsoft/beit-base-patch16-224
|
||
```python
|
||
modelscope download --model microsoft/beit-base-patch16-224 README.md --local_dir /mnt/contest_ceph/zhoushasha/models/microsoft/beit_base_patch16_224_pt22k_ft22k
|
||
```
|
||
|
||
2、使用Dockerfile生成镜像
|
||
联系天数智芯-智铠厂商,获取基础镜像corex:4.3.0
|
||
使用 Dockerfile 生成 镜像
|
||
```python
|
||
docker build -f Dockerfile -t mr100-my:v3 .
|
||
```
|
||
注意 Dockerfile 中已预先将模型 microsoft_beit_base_patch16_224_pt22k_ft22k 放在了 /model 下面
|
||
|
||
3、启动docker
|
||
```python
|
||
docker run -it \
|
||
-p 10086:80 \
|
||
--device=/dev/iluvatar0:/dev/iluvatar0 \
|
||
--name mr_translation_test \
|
||
mr100-my:v3
|
||
```
|
||
|
||
4、测试服务
|
||
```python
|
||
curl -X POST http://localhost:10086/v1/private/s782b4996 \
|
||
> -F "image=@/home/zhoushasha/models/026_0010.jpg"
|
||
```
|