Skip to content

Commit 5d3d544

Browse files
authored
PT 2.6 Inference ARM64 patch (#5622)
* PT 2.6 Inference ARM64 patch
1 parent 304df58 commit 5d3d544

File tree

4 files changed

+7
-4
lines changed

4 files changed

+7
-4
lines changed

pytorch/inference/buildspec-arm64-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: arm64
8-
autopatch_build: "True"
8+
# autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY

pytorch/inference/buildspec-arm64-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: arm64
8-
autopatch_build: "True"
8+
# autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,9 @@ RUN pip install --no-cache-dir \
154154
urllib3 \
155155
idna \
156156
tqdm \
157-
requests
157+
requests \
158+
pytz \
159+
tzdata
158160

159161
# Install PyTorch
160162
RUN pip install --no-cache-dir -U \

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,8 @@ RUN pip install --no-cache-dir \
184184
urllib3 \
185185
idna \
186186
tqdm \
187-
requests
187+
requests \
188+
tzdata
188189

189190
# Install PyTorch
190191
RUN pip install --no-cache-dir -U \

0 commit comments

Comments
 (0)