diff --git a/Dockerfile b/Dockerfile index 39a4e3c..82fa0e1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -15,6 +15,7 @@ ENV PYTHONUNBUFFERED=1 \ COPY requirements.txt /app/requirements.txt COPY packages /app/packages RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ + && python3 -m pip install --upgrade pip \ && pip install --no-cache-dir -r requirements.txt \ && pip uninstall -y onnxruntime onnxruntime-gpu \ && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/