Skip to content

Commit 70cb645

Browse files
authored
PT 2.6 Inference images patch (#5618)
* PT 2.6 Inference images patch
1 parent d0f4103 commit 70cb645

File tree

4 files changed

+7
-5
lines changed

4 files changed

+7
-5
lines changed

pytorch/inference/buildspec-2-6-ec2.yml

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

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY

pytorch/inference/buildspec-2-6-sm.yml

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

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY

pytorch/inference/docker/2.6/py3/Dockerfile.cpu

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,13 @@ RUN pip install --no-cache-dir \
166166
idna \
167167
tqdm \
168168
requests \
169-
setuptools \
169+
"setuptools>=80.10.1" \
170170
urllib3 \
171171
awscli \
172172
types-python-dateutil \
173-
opencv-python
173+
opencv-python \
174+
pytz \
175+
tzdata
174176

175177
# Install PyTorch
176178
RUN pip install --no-cache-dir -U \

pytorch/inference/docker/2.6/py3/cu124/Dockerfile.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ RUN pip install --no-cache-dir -U \
221221
idna \
222222
tqdm \
223223
requests \
224-
setuptools \
224+
"setuptools>=80.10.1" \
225225
urllib3 \
226226
awscli \
227227
types-python-dateutil \

0 commit comments

Comments
 (0)