Skip to content

Commit 5ca9f18

Browse files
Yadan-WeiYadan Wei
andauthored
Build PT27 training with DLC Base (#5171)
* build pt27 * address test error * update cudnn path * add cv2 system lib * adjust dockerfile and change base image * fix oss * address test failure * build with python ARG * fix package regression * regression fix and enable all tests * build sm image * fix buildspec * address sm image test failure * change base image * enable all tests * make cudnn check backward compatible * build ec2 with replaced base uri * build sm image with replaced uri * test ec2 * run sm test * revert toml * revert toml * revert unused change' ' --------- Co-authored-by: Yadan Wei <[email protected]>
1 parent 73f33fe commit 5ca9f18

11 files changed

+163
-360
lines changed

pytorch/training/buildspec-2-7-ec2.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
55
version: &VERSION 2.7.1
66
short_version: &SHORT_VERSION "2.7"
77
arch_type: x86
8-
autopatch_build: "True"
8+
# autopatch_build: "True"
99

1010
repository_info:
1111
training_repository: &TRAINING_REPOSITORY
@@ -36,6 +36,9 @@ context:
3636
deep_learning_container:
3737
source: ../../src/deep_learning_container.py
3838
target: deep_learning_container.py
39+
oss_compliance:
40+
source: ../../scripts/setup_oss_compliance.sh
41+
target: setup_oss_compliance.sh
3942

4043
images:
4144
BuildEC2CPUPTTrainPy3DockerImage:

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,9 @@ context:
3636
deep_learning_container:
3737
source: ../../src/deep_learning_container.py
3838
target: deep_learning_container.py
39+
oss_compliance:
40+
source: ../../scripts/setup_oss_compliance.sh
41+
target: setup_oss_compliance.sh
3942

4043
images:
4144
BuildSageMakerCPUPTTrainPy3DockerImage:

pytorch/training/docker/2.7/py3/Dockerfile.ec2.cpu.core_packages.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@
2323
"skip": "True"
2424
},
2525
"s3torchconnector": {
26-
"version_specifier": "==1.4.2",
26+
"version_specifier": "==1.4.3",
2727
"skip": "True"
2828
},
2929
"accelerate": {
30-
"version_specifier": "==1.9.0",
30+
"version_specifier": "==1.10.1",
3131
"skip": "True"
3232
},
3333
"thinc": {

pytorch/training/docker/2.7/py3/Dockerfile.sagemaker.cpu.core_packages.json

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
"skip": "True"
2424
},
2525
"accelerate": {
26-
"version_specifier": "==1.10.0",
26+
"version_specifier": "==1.10.1",
2727
"skip": "True"
2828
},
2929
"thinc": {
@@ -64,8 +64,5 @@
6464
},
6565
"opencv-python": {
6666
"version_specifier": "==4.11.0.86"
67-
},
68-
"sagemaker-pytorch-training": {
69-
"version_specifier": ">=2.9.0,<3"
7067
}
7168
}

pytorch/training/docker/2.7/py3/Dockerfile.sagemaker.cpu.py_scan_allowlist.json

Lines changed: 0 additions & 3 deletions
This file was deleted.

pytorch/training/docker/2.7/py3/cu128/Dockerfile.ec2.gpu.core_packages.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@
2323
"skip": "True"
2424
},
2525
"s3torchconnector": {
26-
"version_specifier": "==1.4.2",
26+
"version_specifier": "==1.4.3",
2727
"skip": "True"
2828
},
2929
"accelerate": {
30-
"version_specifier": "==1.9.0",
30+
"version_specifier": "==1.10.1",
3131
"skip": "True"
3232
},
3333
"thinc": {

0 commit comments

Comments
 (0)