From 0afda8b058447d621f0c90bf89cdbb001ba63f6d Mon Sep 17 00:00:00 2001 From: Duyi-Wang Date: Tue, 9 Jul 2024 10:41:00 +0800 Subject: [PATCH] [xFasterTransformer] Add procps-ng and symlink for `python`. --- AI/xfastertransformer/1.7.3/23/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/AI/xfastertransformer/1.7.3/23/Dockerfile b/AI/xfastertransformer/1.7.3/23/Dockerfile index 6b32330..43fde26 100644 --- a/AI/xfastertransformer/1.7.3/23/Dockerfile +++ b/AI/xfastertransformer/1.7.3/23/Dockerfile @@ -7,7 +7,8 @@ LABEL \ org.opencontainers.image.version="1.7.3-23" RUN dnf install anolis-epao-release -y -RUN dnf install -y wget numactl python3-pip git vim level-zero-devel +RUN dnf install -y wget numactl python3-pip git vim level-zero-devel procps-ng +RUN ln /usr/bin/python3 /usr/bin/python RUN pip3 install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu RUN pip3 install --no-cache-dir transformers==4.41.2 sentencepiece==0.1.99 accelerate==0.23.0 protobuf==4.25.3 tiktoken==0.6.0 RUN pip3 install --no-cache-dir xfastertransformer==1.7.3 -- Gitee