From 263728066720efccb1fc83bb210ff809169fbdc5 Mon Sep 17 00:00:00 2001 From: kevincheng2 Date: Mon, 2 Sep 2024 13:55:37 +0800 Subject: [PATCH] update dockerfile --- llm/dockerfiles/Dockerfile_serving_cuda118_cudnn8 | 4 ++-- llm/dockerfiles/Dockerfile_serving_cuda123_cudnn9 | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/llm/dockerfiles/Dockerfile_serving_cuda118_cudnn8 b/llm/dockerfiles/Dockerfile_serving_cuda118_cudnn8 index 157353fb7b..3f5a2a511f 100644 --- a/llm/dockerfiles/Dockerfile_serving_cuda118_cudnn8 +++ b/llm/dockerfiles/Dockerfile_serving_cuda118_cudnn8 @@ -6,10 +6,10 @@ COPY ./client/ /opt/output/client/ ENV LD_LIBRARY_PATH "/usr/local/cuda-11.8/compat/:$LD_LIBRARY_PATH" +RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple RUN python3 -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu118/ \ && python3 -m pip install paddlenlp==3.0.0b0 \ - && python3 -m pip install --no-cache-dir sentencepiece pycryptodome tritonclient[all]==2.41.1 \ - && apt-get clean && rm -rf /var/lib/apt/lists/* + && python3 -m pip install --no-cache-dir sentencepiece pycryptodome tritonclient[all]==2.41.1 RUN git clone https://gitee.com/paddlepaddle/PaddleNLP.git && cd PaddleNLP/csrc \ && python3 setup_cuda.py build && python3 setup_cuda.py install --user \ diff --git a/llm/dockerfiles/Dockerfile_serving_cuda123_cudnn9 b/llm/dockerfiles/Dockerfile_serving_cuda123_cudnn9 index 1ab89d6ec5..786bf2aecd 100644 --- a/llm/dockerfiles/Dockerfile_serving_cuda123_cudnn9 +++ b/llm/dockerfiles/Dockerfile_serving_cuda123_cudnn9 @@ -6,10 +6,10 @@ COPY ./client/ /opt/output/client/ ENV LD_LIBRARY_PATH "/usr/local/cuda-12.3/compat/:$LD_LIBRARY_PATH" +RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple RUN python3 -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu123/ \ && python3 -m pip install paddlenlp==3.0.0b0 \ - && python3 -m pip install --no-cache-dir sentencepiece pycryptodome tritonclient[all]==2.41.1 \ - && apt-get clean && rm -rf /var/lib/apt/lists/* + && python3 -m pip install --no-cache-dir sentencepiece pycryptodome tritonclient[all]==2.41.1 RUN git clone https://gitee.com/paddlepaddle/PaddleNLP.git && cd PaddleNLP/csrc \ && python3 setup_cuda.py build && python3 setup_cuda.py install --user \