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.
azavea · Jan 23, 2024 · fe37b51 · fe37b51
1 parent 6a918ba
commit fe37b51
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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