diff --git a/pytorch/jobs/docker/2.2/py3/Dockerfile.gpu b/pytorch/jobs/docker/2.2/py3/Dockerfile.gpu index 426c225..fdfd6cf 100644 --- a/pytorch/jobs/docker/2.2/py3/Dockerfile.gpu +++ b/pytorch/jobs/docker/2.2/py3/Dockerfile.gpu @@ -77,8 +77,8 @@ RUN ${PIP} install --no-cache --upgrade \ -r requirements.txt # install cuQuantum -RUN wget https://developer.download.nvidia.com/compute/cuquantum/redist/cuquantum/linux-x86_64/cuquantum-linux-x86_64-23.03.0.20-archive.tar.xz && tar xvf ./cuquantum-linux-x86_64-23.03.0.20-archive.tar.xz -RUN mv /cuquantum-linux-x86_64-23.03.0.20-archive /opt/cuquantum +RUN wget https://developer.download.nvidia.com/compute/cuquantum/redist/cuquantum/linux-x86_64/cuquantum-linux-x86_64-24.03.0.4_cuda12-archive.tar.xz && tar xvf ./cuquantum-linux-x86_64-24.03.0.4_cuda12-archive.tar.xz +RUN mv /cuquantum-linux-x86_64-24.03.0.4_cuda12-archive /opt/cuquantum ENV LD_LIBRARY_PATH="/opt/cuquantum/lib:${LD_LIBRARY_PATH}" diff --git a/tensorflow/jobs/docker/2.14/py3/requirements.txt b/tensorflow/jobs/docker/2.14/py3/requirements.txt index 5388e9e..21a05ae 100644 --- a/tensorflow/jobs/docker/2.14/py3/requirements.txt +++ b/tensorflow/jobs/docker/2.14/py3/requirements.txt @@ -1,10 +1,10 @@ -amazon-braket-default-simulator==1.23.0 +amazon-braket-default-simulator==1.23.1 amazon-braket-schemas==1.21.4 -amazon-braket-pennylane-plugin==1.25.0 +amazon-braket-pennylane-plugin==1.26.0 amazon-braket-sdk==1.78.0 -awscli==1.32.89 -botocore==1.34.89 -boto3==1.34.89 +awscli==1.32.94 +botocore==1.34.94 +boto3==1.34.94 dask==2024.4.2 h5py==3.11.0 ipykernel==6.29.4 @@ -17,7 +17,7 @@ openfermion==1.6.1 pandas==2.2.2 PennyLane==0.35.1 PennyLane-Lightning==0.35.1 -pydantic==2.7.0 +pydantic==2.7.1 requests==2.31.0 sagemaker-training==4.7.4 scikit-learn==1.4.2