Browse Source

fix: update Ascend.md and npu.Dockerfile for improved clarity on Docker image tags and usage instructions

myhloli 1 week ago
parent
commit
669b6cd629
2 changed files with 9 additions and 9 deletions
  1. 2 5
      docker/china/npu.Dockerfile
  2. 7 4
      docs/zh/usage/acceleration_cards/Ascend.md

+ 2 - 5
docker/china/npu.Dockerfile

@@ -1,12 +1,9 @@
 # 基础镜像配置
 # Base image containing the LMDeploy inference environment, requiring ARM(AArch64) CPU + Ascend NPU.
-FROM crpi-4crprmm5baj1v8iv.cn-hangzhou.personal.cr.aliyuncs.com/lmdeploy_dlinfer/ascend:mineru-a2 AS lmdeploy
+FROM crpi-4crprmm5baj1v8iv.cn-hangzhou.personal.cr.aliyuncs.com/lmdeploy_dlinfer/ascend:mineru-a2
 # Base image containing the vLLM inference environment, requiring ARM(AArch64) CPU + Ascend NPU.
-FROM quay.io/ascend/vllm-ascend:v0.11.0rc1 AS vllm
+# FROM quay.io/ascend/vllm-ascend:v0.11.0rc1
 
-# 默认使用 lmdeploy
-ARG INFERENCE_ENGINE=lmdeploy
-FROM ${INFERENCE_ENGINE} AS final
 
 # Install libgl for opencv support & Noto fonts for Chinese characters
 RUN apt-get update && \

+ 7 - 4
docs/zh/usage/acceleration_cards/Ascend.md

@@ -17,17 +17,17 @@ Ascend加速卡支持使用`lmdeploy`或`vllm`进行VLM模型推理加速。请
 
 ```bash
 wget https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/docker/china/npu.Dockerfile
-docker build --build-arg INFERENCE_ENGINE=lmdeploy -t mineru:latest -f npu.Dockerfile .
+docker build --build-arg INFERENCE_ENGINE=lmdeploy -t mineru:lmdeploy-latest -f npu.Dockerfile .
 ```
 
-
 ### 2.2 使用 vllm
 
 #### 使用 Dockerfile 构建镜像
 
 ```bash
 wget https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/docker/china/npu.Dockerfile
-docker build --build-arg INFERENCE_ENGINE=vllm -t mineru:latest -f npu.Dockerfile .
+# 下载 Dockerfile后,使用编辑器打开并注释掉第3行的基础镜像,打开第5行的基础镜像后再执行后续操作
+docker build --build-arg INFERENCE_ENGINE=vllm -t mineru:vllm-latest -f npu.Dockerfile .
 ``` 
 
 ## 3. 启动 Docker 容器
@@ -52,9 +52,12 @@ docker run -u root --name mineru_docker --privileged=true \
     -v /usr/local/dcmi:/usr/local/dcmi \
     -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
     -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
-    -itd mineru:latest \
+    -itd mineru:lmdeploy-latest \
     /bin/bash
 ```
 
+>[!TIP]
+> 请根据实际情况选择使用`lmdeploy`或`vllm`版本的镜像,替换上述命令中的`mineru:lmdeploy-latest`为`mineru:vllm-latest`即可。
+
 执行该命令后,您将进入到Docker容器的交互式终端,并映射了一些端口用于可能会使用的服务,您可以直接在容器内运行MinerU相关命令来使用MinerU的功能。
 您也可以直接通过替换`/bin/bash`为服务启动命令来启动MinerU服务,详细说明请参考[通过命令启动服务](https://opendatalab.github.io/MinerU/zh/usage/quick_usage/#apiwebuihttp-clientserver)。