infra: upgrading containers (amazon-braket#186)

.github/dependabot.yml

@@ -47,7 +47,7 @@ updates:
 
   - package-ecosystem: "pip"
     # Files stored in repository for the tensorflow image
-    directory: "/tensorflow/jobs/docker/2.13/py3/"
+    directory: "/tensorflow/jobs/docker/2.14/py3/"
     schedule:
       interval: "weekly"
     commit-message:
@@ -64,7 +64,7 @@ updates:
 
   - package-ecosystem: "pip"
     # Files stored in repository for the pytorch image
-    directory: "/pytorch/jobs/docker/2.0/py3/"
+    directory: "/pytorch/jobs/docker/2.2/py3/"
     schedule:
       interval: "weekly"
     commit-message:

.github/workflows/build-py-reqs.yml

@@ -17,7 +17,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        path: ["base/jobs/docker/1.0/py3/requirements.txt", "pytorch/jobs/docker/2.1/py3/requirements.txt", "tensorflow/jobs/docker/2.13/py3/requirements.txt"]
+        path: ["base/jobs/docker/1.0/py3/requirements.txt", "pytorch/jobs/docker/2.2/py3/requirements.txt", "tensorflow/jobs/docker/2.14/py3/requirements.txt"]
     steps:
       - uses: actions/checkout@v4
       - name: setup python

pytorch/buildspec.yml

@@ -1,8 +1,8 @@
 account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
 region: &REGION <set-$REGION-in-environment>
 framework: &FRAMEWORK pytorch
-version: &VERSION 2.1.0
-short_version: &SHORT_VERSION 2.1
+version: &VERSION 2.2.0
+short_version: &SHORT_VERSION 2.2
 
 repository_info:
   jobs_repository: &JOBS_REPOSITORY
@@ -20,7 +20,7 @@ context:
       source: ../../src/braket_container.py
       target: braket_container.py
     requirements:
-      source: ../../pytorch/jobs/docker/2.1/py3/requirements.txt
+      source: ../../pytorch/jobs/docker/2.2/py3/requirements.txt
       target: requirements.txt
 
 images:

pytorch/jobs/docker/2.1/py3/Dockerfile.gpu → pytorch/jobs/docker/2.2/py3/Dockerfile.gpu

@@ -1,7 +1,7 @@
 # This is based loosely on :
 # https://github.com/aws/amazon-sagemaker-examples/blob/master/advanced_functionality/custom-training-containers/script-mode-container/docker/Dockerfile
 
-FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu121-ubuntu20.04-ec2
+FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-ec2
 
 LABEL maintainer="Amazon Braket"
 LABEL major_version="1"

tensorflow/buildspec.yml

@@ -1,8 +1,8 @@
 account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
 region: &REGION <set-$REGION-in-environment>
 framework: &FRAMEWORK tensorflow
-version: &VERSION 2.13.0
-short_version: &SHORT_VERSION 2.13
+version: &VERSION 2.14.1
+short_version: &SHORT_VERSION 2.14
 
 repository_info:
   jobs_repository: &JOBS_REPOSITORY
@@ -20,7 +20,7 @@ context:
       source: ../../src/braket_container.py
       target: braket_container.py
     requirements:
-      source: ../../tensorflow/jobs/docker/2.13/py3/requirements.txt
+      source: ../../tensorflow/jobs/docker/2.14/py3/requirements.txt
       target: requirements.txt
 
 images:

tensorflow/jobs/docker/2.13/py3/Dockerfile.gpu → tensorflow/jobs/docker/2.14/py3/Dockerfile.gpu

@@ -1,6 +1,6 @@
 # Start with AWS Deep Learning Container. See
 # https://github.com/aws/deep-learning-containers/blob/master/available_images.md
-FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.13.0-gpu-py310-cu118-ubuntu20.04-ec2
+FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.14.1-gpu-py310-cu118-ubuntu20.04-ec2
 
 LABEL maintainer="Amazon Braket"
 LABEL major_version="1"