Эх сурвалжийг харах

feat: add Dockerfiles for camb and maca environments, update ppu base image

myhloli 2 долоо хоног өмнө
parent
commit
4ba9c73458

+ 23 - 0
docker/china/camb.Dockerfile

@@ -0,0 +1,23 @@
+# Base image containing the LMDeploy inference environment, requiring amd64 CPU + cambricon MLU.
+FROM
+
+# Install libgl for opencv support & Noto fonts for Chinese characters
+RUN apt-get update && \
+    apt-get install -y \
+        fonts-noto-core \
+        fonts-noto-cjk \
+        fontconfig \
+        libgl1 && \
+    fc-cache -fv && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install mineru latest
+RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
+    python3 -m pip cache purge
+
+# Download models and update the configuration file
+RUN /bin/bash -c "mineru-models-download -s modelscope -m all"
+
+# Set the entry point to activate the virtual environment and run the command line tool
+ENTRYPOINT ["/bin/bash", "-c", "export MINERU_MODEL_SOURCE=local && exec \"$@\"", "--"]

+ 23 - 0
docker/china/maca.Dockerfile

@@ -0,0 +1,23 @@
+# Base image containing the LMDeploy inference environment, requiring amd64 CPU + metax GPU.
+FROM
+
+# Install libgl for opencv support & Noto fonts for Chinese characters
+RUN apt-get update && \
+    apt-get install -y \
+        fonts-noto-core \
+        fonts-noto-cjk \
+        fontconfig \
+        libgl1 && \
+    fc-cache -fv && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install mineru latest
+RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
+    python3 -m pip cache purge
+
+# Download models and update the configuration file
+RUN /bin/bash -c "mineru-models-download -s modelscope -m all"
+
+# Set the entry point to activate the virtual environment and run the command line tool
+ENTRYPOINT ["/bin/bash", "-c", "export MINERU_MODEL_SOURCE=local && exec \"$@\"", "--"]

+ 1 - 1
docker/china/ppu.Dockerfile

@@ -1,4 +1,4 @@
-# Base image containing the LMDeploy inference environment, requiring amd64 CPU + ppu.
+# Base image containing the LMDeploy inference environment, requiring amd64 CPU + t-head PPU.
 FROM crpi-4crprmm5baj1v8iv.cn-hangzhou.personal.cr.aliyuncs.com/lmdeploy_dlinfer/ppu:mineru-ppu
 
 # Install libgl for opencv support & Noto fonts for Chinese characters