Skip to content

Commit dd18cad

Browse files
authored
Enable PT 2.4 Training Autopatch (#4462)
* Enable PT 2.4 Training Autopatch * revert toml
1 parent 4d41082 commit dd18cad

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pytorch/training/buildspec-2-4-sm.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
55
version: &VERSION 2.4.0
66
short_version: &SHORT_VERSION "2.4"
77
arch_type: x86
8-
autopatch_build: "False"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
training_repository: &TRAINING_REPOSITORY
@@ -41,7 +41,7 @@ images:
4141
tag_python_version: &TAG_PYTHON_VERSION py311
4242
os_version: &OS_VERSION ubuntu22.04
4343
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
44-
# latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
44+
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
4545
# build_tag_override: "beta:2.4.0-cpu-py311-ubuntu22.04-sagemaker"
4646
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ]
4747
target: sagemaker
@@ -57,7 +57,7 @@ images:
5757
cuda_version: &CUDA_VERSION cu124
5858
os_version: &OS_VERSION ubuntu22.04
5959
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
60-
# latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
60+
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sagemaker" ]
6161
# build_tag_override: "beta:2.4.0-gpu-py311-cu124-ubuntu22.04-sagemaker"
6262
docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION, /Dockerfile.,
6363
*DEVICE_TYPE ]

0 commit comments

Comments
 (0)