From fb3b199189f40b036a8ebec52642f4a3b795e360 Mon Sep 17 00:00:00 2001 From: Adeel Hassan Date: Fri, 22 Nov 2024 21:47:21 -0500 Subject: [PATCH] install onnxruntime from pypi to avoid installing numpy 2 --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index bc0c78580..022527bd1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -165,7 +165,7 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main # --extra-index-url in requirements.txt seems to cause problems with the # RTD build. RUN if [ "${TARGETARCH}" != "arm64" ]; then \ - uv pip install --upgrade onnxruntime-gpu==1.17 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/; fi + uv pip install onnxruntime-gpu==1.19; fi #------------------------------------------------------------------------