From 7fc72b8eed463ceceba4889235f1ff9ae3d95b72 Mon Sep 17 00:00:00 2001 From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com> Date: Fri, 22 Nov 2024 22:52:04 +0000 Subject: [PATCH 1/2] Bump albumentations in /rastervision_pytorch_learner Bumps [albumentations](https://github.com/albumentations-team/albumentations) from 1.4.13 to 1.4.21. - [Release notes](https://github.com/albumentations-team/albumentations/releases) - [Commits](https://github.com/albumentations-team/albumentations/compare/1.4.13...1.4.21) --- updated-dependencies: - dependency-name: albumentations dependency-type: direct:production update-type: version-update:semver-patch ... Signed-off-by: dependabot[bot] --- rastervision_pytorch_learner/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rastervision_pytorch_learner/requirements.txt b/rastervision_pytorch_learner/requirements.txt index 2509d98cf..e8fdb1ffe 100644 --- a/rastervision_pytorch_learner/requirements.txt +++ b/rastervision_pytorch_learner/requirements.txt @@ -5,7 +5,7 @@ pillow==10.4.0 torch==2.4.0 torchvision==0.19.0 tensorboard==2.17.0 -albumentations==1.4.13 +albumentations==1.4.21 cython==3.0.11 pycocotools==2.0.8 psutil==6.0.0 From b62526b1e21cad095f13179fadf76664aa8cd428 Mon Sep 17 00:00:00 2001 From: Adeel Hassan Date: Fri, 22 Nov 2024 21:47:21 -0500 Subject: [PATCH 2/2] 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 #------------------------------------------------------------------------