diff --git a/base/buildspec.yml b/base/buildspec.yml index 09e0796e..138bc6f4 100644 --- a/base/buildspec.yml +++ b/base/buildspec.yml @@ -1,7 +1,7 @@ account_id: &ACCOUNT_ID region: ®ION framework: &FRAMEWORK base -version: &VERSION 1.0 +version: &VERSION 1.1 repository_info: jobs_repository: &JOBS_REPOSITORY @@ -27,7 +27,7 @@ images: device_type: &DEVICE_TYPE cpu python_version: &DOCKER_PYTHON_VERSION py3 tag_python_version: &TAG_PYTHON_VERSION py37 - os_version: &OS_VERSION ubuntu18.04 + os_version: &OS_VERSION ubuntu22.04 tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION ] docker_file: !join [ docker/, *VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ] context: diff --git a/base/jobs/docker/1.0/py3/Dockerfile.cpu b/base/jobs/docker/1.0/py3/Dockerfile.cpu index 9acc1dbd..f171aea5 100644 --- a/base/jobs/docker/1.0/py3/Dockerfile.cpu +++ b/base/jobs/docker/1.0/py3/Dockerfile.cpu @@ -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 public.ecr.aws/ubuntu/ubuntu:18.04 +FROM public.ecr.aws/lts/ubuntu:22.04 LABEL maintainer="Amazon Braket" LABEL major_version="1" @@ -16,10 +16,10 @@ ENV PYTHONIOENCODING=UTF-8 ENV LANG=C.UTF-8 ENV LC_ALL=C.UTF-8 -ARG PYTHON=python3.7 +ARG PYTHON=python3.10 ARG PYTHON_PIP=python3-pip ARG PIP=pip3 -ARG PYTHON_VERSION=3.7.10 +ARG PYTHON_VERSION=3.10.11 RUN apt-get update \ && apt-get install -y --no-install-recommends \ @@ -106,25 +106,24 @@ RUN apt-get update && apt-get -y install cmake protobuf-compiler # Installing our custom libraries RUN ${PIP} install --no-cache --upgrade \ - amazon-braket-default-simulator==1.3.0 \ - amazon-braket-ocean-plugin==1.0.6 \ - amazon-braket-schemas==1.8.0 \ - amazon-braket-sdk==1.18.0 \ - awscli==1.22.15 \ - botocore==1.23.15 \ - boto3==1.20.15 \ - decorator==4.4.0 \ - dask==2.30.0 \ - dwave-ocean-sdk==4.2.0 \ - matplotlib==3.2.2 \ - numpy==1.21.4 \ - pandas==1.1.4 \ - rsa==4.4.1 \ - scikit-learn==0.20.3 \ + amazon-braket-default-simulator==1.14.0 \ + amazon-braket-schemas==1.17.0 \ + amazon-braket-pennylane-plugin==1.15.2 \ + amazon-braket-sdk==1.39.0 \ + awscli==1.27.65 \ + botocore==1.29.65 \ + boto3==1.26.64 \ + 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 \ + scikit-learn==1.2.2 \ requests==2.26.0 \ - scipy==1.7.3 + scipy==1.9.3 -RUN ${PIP} install --no-cache --upgrade sagemaker-training +RUN ${PIP} install --no-cache --upgrade sagemaker-training==4.4.10 RUN HOME_DIR=/root \ && curl -o ${HOME_DIR}/oss_compliance.zip https://aws-dlinfra-utilities.s3.amazonaws.com/oss_compliance.zip \