feat: upgrade to tensorflow 2.16

amazon-braket · Oct 17, 2024 · 1af06f2 · 1af06f2
1 parent 5cffdd7
commit 1af06f2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tensorflow/buildspec.yml b/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.14.1
-short_version: &SHORT_VERSION 2.14
+version: &VERSION 2.16.2
+short_version: &SHORT_VERSION 2.16
 
 repository_info:
   jobs_repository: &JOBS_REPOSITORY
@@ -32,7 +32,7 @@ images:
     python_version: &DOCKER_PYTHON_VERSION py3
     tag_python_version: &TAG_PYTHON_VERSION py310
     os_version: &OS_VERSION ubuntu20.04
-    cuda_version: &CUDA_VERSION cu118
+    cuda_version: &CUDA_VERSION cu123
     tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION ]
     docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ]
     context:

diff --git a/...rflow/jobs/docker/2.14/py3/Dockerfile.gpu → ...rflow/jobs/docker/2.16/py3/Dockerfile.gpu b/...rflow/jobs/docker/2.14/py3/Dockerfile.gpu → ...rflow/jobs/docker/2.16/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.14.1-gpu-py310-cu118-ubuntu20.04-ec2
+FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.16.2-gpu-py310-cu123-ubuntu20.04-ec2
 
 LABEL maintainer="Amazon Braket"
 LABEL major_version="1"

diff --git a/...low/jobs/docker/2.14/py3/requirements.txt → ...low/jobs/docker/2.16/py3/requirements.txt b/...low/jobs/docker/2.14/py3/requirements.txt → ...low/jobs/docker/2.16/py3/requirements.txt