From b4fd32ac74f213de9a7620500d415d540ab27960 Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Mon, 8 Sep 2025 18:13:58 +0800
Subject: [PATCH] update readme

---
 Dockerfile.ascend => Dockerfile |  2 +-
 README.md                       | 23 +++++++++++++++++++++++
 run_in_docker.sh                |  4 ++++
 3 files changed, 28 insertions(+), 1 deletion(-)
 rename Dockerfile.ascend => Dockerfile (78%)
 create mode 100644 README.md
 create mode 100755 run_in_docker.sh

diff --git a/Dockerfile.ascend b/Dockerfile
similarity index 78%
rename from Dockerfile.ascend
rename to Dockerfile
index 9ed5905..d988cdc 100644
--- a/Dockerfile.ascend
+++ b/Dockerfile
@@ -3,4 +3,4 @@ FROM git.modelhub.org.cn:9443/enginex-ascend/vllm-ascend:v0.10.0rc1
 
 WORKDIR /workspace
 RUN pip install sentence-transformers
-COPY main.py dataset.json /workspace/
+COPY main.py test.sh dataset.json /workspace/
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..03df275
--- /dev/null
+++ b/README.md
@@ -0,0 +1,23 @@
+## Quickstart
+
+### 构建镜像
+```bash
+docker build -t feature:v0.1 .
+```
+
+### 模型下载
+模型地址：https://modelscope.cn/models/BAAI/bge-large-zh-v1.5
+并放到目录：`/mnt/contest_ceph/zhanghao/models/BAAI/bge-large-zh-v1.5`（如更改目录，请修改后面的执行脚本中的模型路径）
+
+### 测试程序
+1. 准备输入数据集，可以参考示例`dataset.json`
+2. 在docker镜像里运行测试程序，会根据`dataset.json`内容，计算每个句子的embedding，同时计算所有句子的两两相似度，结果保存在`output.json`。
+```bash
+./run_in_docker.sh
+```
+
+## 测试结果
+|      | A100 平均生成时间（秒） | 昇腾910B 平均生成时间（秒） |
+|------|-------------------------|----------------------------|
+| 时间 | 0.0032                  | 0.0138                     |
+
diff --git a/run_in_docker.sh b/run_in_docker.sh
new file mode 100755
index 0000000..4ad8062
--- /dev/null
+++ b/run_in_docker.sh
@@ -0,0 +1,4 @@
+#! /usr/bin/env bash
+image=feature:v0.1
+device=1
+docker run -v `pwd`:/workspace -e ASCEND_VISIBLE_DEVICES=$device -e NPU_VISIBLE_DEVICES=${device} --device /dev/davinci$device:/dev/davinci0 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc -v /mnt:/mnt -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info -v /etc/ascend_install.info:/etc/ascend_install.info --privileged --entrypoint bash  $image test.sh