diff --git a/pytorch/inference/buildspec-2-5-ec2.yml b/pytorch/inference/buildspec-2-5-ec2.yml index a20ca9443cdb..a34fb55735f8 100644 --- a/pytorch/inference/buildspec-2-5-ec2.yml +++ b/pytorch/inference/buildspec-2-5-ec2.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.5.1 short_version: &SHORT_VERSION "2.5" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY diff --git a/pytorch/inference/buildspec-2-5-sm.yml b/pytorch/inference/buildspec-2-5-sm.yml index 354d31f305a4..0bba3d6138f4 100644 --- a/pytorch/inference/buildspec-2-5-sm.yml +++ b/pytorch/inference/buildspec-2-5-sm.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.5.1 short_version: &SHORT_VERSION "2.5" arch_type: x86 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY diff --git a/pytorch/inference/buildspec.yml b/pytorch/inference/buildspec.yml index 767e9a8e9bab..16412edfa389 100644 --- a/pytorch/inference/buildspec.yml +++ b/pytorch/inference/buildspec.yml @@ -1 +1 @@ -buildspec_pointer: buildspec-2-5-ec2.yml +buildspec_pointer: buildspec-2-5-sm.yml diff --git a/pytorch/inference/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json b/pytorch/inference/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json new file mode 100644 index 000000000000..b0b255fda778 --- /dev/null +++ b/pytorch/inference/docker/2.5/py3/Dockerfile.ec2.cpu.core_packages.json @@ -0,0 +1,27 @@ +{ + "captum": { + "version_specifier": "==0.6.0", + "skip": "True" + }, + "torchaudio": { + "version_specifier": "==2.5.1+cpu", + "skip": "True" + }, + "torchtext": { + "version_specifier": "==0.18.0+cpu", + "skip": "True" + }, + "torchvision": { + "version_specifier": "==0.20.1+cpu", + "skip": "True" + }, + "torchserve": { + "version_specifier": "==0.12.0" + }, + "requests": { + "version_specifier": ">=2.32.3" + }, + "torch-model-archiver": { + "version_specifier": "==0.12.0" + } +} diff --git a/pytorch/inference/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json b/pytorch/inference/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json new file mode 100644 index 000000000000..3856bf997ccc --- /dev/null +++ b/pytorch/inference/docker/2.5/py3/Dockerfile.sagemaker.cpu.core_packages.json @@ -0,0 +1,30 @@ +{ + "captum": { + "version_specifier": "==0.6.0", + "skip": "True" + }, + "torchaudio": { + "version_specifier": "==2.5.1+cpu", + "skip": "True" + }, + "torchtext": { + "version_specifier": "==0.18.0+cpu", + "skip": "True" + }, + "torchvision": { + "version_specifier": "==0.20.1+cpu", + "skip": "True" + }, + "requests": { + "version_specifier": ">=2.32.3" + }, + "torchserve": { + "version_specifier": "==0.12.0" + }, + "torch-model-archiver": { + "version_specifier": "==0.12.0" + }, + "sagemaker-pytorch-inference": { + "version_specifier": "==2.0.25" + } +} diff --git a/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json b/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json new file mode 100644 index 000000000000..25976b9498cf --- /dev/null +++ b/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.ec2.gpu.core_packages.json @@ -0,0 +1,27 @@ +{ + "captum": { + "version_specifier": "==0.6.0", + "skip": "True" + }, + "torchaudio": { + "version_specifier": "==2.5.1+cu124", + "skip": "True" + }, + "torchtext": { + "version_specifier": "==0.18.0+cu124", + "skip": "True" + }, + "torchvision": { + "version_specifier": "==0.20.1+cu124", + "skip": "True" + }, + "requests": { + "version_specifier": ">=2.32.3" + }, + "torchserve": { + "version_specifier": "==0.12.0" + }, + "torch-model-archiver": { + "version_specifier": "==0.12.0" + } +} diff --git a/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json b/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json new file mode 100644 index 000000000000..0b040c6962f6 --- /dev/null +++ b/pytorch/inference/docker/2.5/py3/cu124/Dockerfile.sagemaker.gpu.core_packages.json @@ -0,0 +1,30 @@ +{ + "captum": { + "version_specifier": "==0.6.0", + "skip": "True" + }, + "torchaudio": { + "version_specifier": "==2.5.1+cu124", + "skip": "True" + }, + "torchtext": { + "version_specifier": "==0.18.0+cu124", + "skip": "True" + }, + "torchvision": { + "version_specifier": "==0.20.1+cu124", + "skip": "True" + }, + "requests": { + "version_specifier": ">=2.32.3" + }, + "torchserve": { + "version_specifier": "==0.12.0" + }, + "torch-model-archiver": { + "version_specifier": "==0.12.0" + }, + "sagemaker-pytorch-inference": { + "version_specifier": "==2.0.25" + } +}