infra: update braket dependencies

amazon-braket · Aug 29, 2023 · 0e43709 · 0e43709
1 parent 0108d8c
commit 0e43709
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 18 deletions.
diff --git a/base/jobs/docker/1.0/py3/Dockerfile.cpu b/base/jobs/docker/1.0/py3/Dockerfile.cpu
@@ -107,19 +107,19 @@ RUN apt-get update && apt-get -y install cmake protobuf-compiler
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
         dask==2023.1.1 \
         matplotlib==3.6.3 \
         numpy==1.23.5 \
         pandas==2.0.0 \
-        pennylane==0.30.0 \
-        PennyLane-Lightning==0.30.0 \
+        pennylane==0.31.0 \
+        PennyLane-Lightning==0.31.0 \
         scikit-learn==1.2.2 \
         requests==2.26.0 \
         scipy==1.9.3

diff --git a/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu b/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu
@@ -73,10 +73,10 @@ RUN apt-get update && apt-get -y install cmake protobuf-compiler
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
@@ -91,8 +91,8 @@ RUN ${PIP} install --no-cache --upgrade \
         numpy==1.23.5 \
         openfermion==1.5.1 \
         pandas==2.0.0 \
-        PennyLane==0.30.0 \
-        PennyLane-Lightning[gpu]==0.30.0 \
+        PennyLane==0.31.0 \
+        PennyLane-Lightning[gpu]==0.31.0 \
         pydantic==1.10.6 \
         requests==2.26.0 \
         scikit-learn==1.2.2 \

diff --git a/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu b/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu
@@ -15,10 +15,10 @@ RUN apt-get update \
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
@@ -32,8 +32,8 @@ RUN ${PIP} install --no-cache --upgrade \
         numpy==1.23.5 \
         openfermion==1.5.1 \
         pandas==2.0.0 \
-        PennyLane==0.30.0 \
-        PennyLane-Lightning==0.30.0 \
+        PennyLane==0.31.0 \
+        PennyLane-Lightning==0.31.0 \
         pydantic==1.10.6 \
         requests==2.26.0 \
         scikit-learn==1.2.2 \