From fe37b511ad14bdced70f6c7294f55df4e2a505ac Mon Sep 17 00:00:00 2001 From: Adeel Hassan Date: Thu, 11 Jan 2024 15:01:43 -0500 Subject: [PATCH] add sagemaker_pytorch_training dependency in Dockerfile - This is required for using the image via AWS SageMaker's PyTorch Estimator. - Also bump boto3 and awscli to compatible versions. --- Dockerfile | 4 ++++ rastervision_aws_batch/requirements.txt | 4 ++-- rastervision_aws_s3/requirements.txt | 4 ++-- rastervision_core/requirements.txt | 2 +- 4 files changed, 9 insertions(+), 5 deletions(-) diff --git a/Dockerfile b/Dockerfile index 63473a4fe..d6c99d1d0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -166,4 +166,8 @@ COPY ./rastervision_aws_sagemaker/ /opt/src/rastervision_aws_sagemaker/ COPY ./rastervision_pytorch_backend/ /opt/src/rastervision_pytorch_backend/ COPY ./rastervision_pytorch_learner/ /opt/src/rastervision_pytorch_learner/ +# needed for this image to be used by the AWS SageMaker PyTorch Estimator +RUN pip install sagemaker_pytorch_training==2.8.1 +ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main + CMD ["bash"] diff --git a/rastervision_aws_batch/requirements.txt b/rastervision_aws_batch/requirements.txt index 6be6833d3..7d12ce8db 100644 --- a/rastervision_aws_batch/requirements.txt +++ b/rastervision_aws_batch/requirements.txt @@ -1,3 +1,3 @@ rastervision_pipeline==0.21.4-dev -boto3==1.28.8 -awscli==1.29.8 +boto3==1.34.14 +awscli==1.32.14 diff --git a/rastervision_aws_s3/requirements.txt b/rastervision_aws_s3/requirements.txt index 35d4e9564..9b9e1e744 100644 --- a/rastervision_aws_s3/requirements.txt +++ b/rastervision_aws_s3/requirements.txt @@ -1,5 +1,5 @@ rastervision_pipeline==0.21.4-dev -boto3==1.28.8 -awscli==1.29.8 +boto3==1.34.14 +awscli==1.32.14 tqdm==4.65.0 diff --git a/rastervision_core/requirements.txt b/rastervision_core/requirements.txt index dd17f8fe2..d36bee652 100644 --- a/rastervision_core/requirements.txt +++ b/rastervision_core/requirements.txt @@ -12,6 +12,6 @@ opencv-python-headless==4.6.0.66 tqdm==4.65.0 xarray==2023.2.0 scikit-image==0.21.0 -boto3==1.28.8 +boto3==1.34.14 stackstac==0.5.0 humanize==4.8.0