Skip to content

Commit 1909bf4

Browse files
authored
add cu126 torchvision and torchaudio aarch64 build (#6062)
related to pytorch/pytorch#138440 cc @atalman @ptrblck @nWEIdia
1 parent 449c379 commit 1909bf4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
163163
XPU: "pytorch/manylinux2_28-builder:xpu",
164164
# TODO: Migrate CUDA_AARCH64 image to manylinux2_28_aarch64-builder:cuda12.4
165165
CPU_AARCH64: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64",
166-
CUDA_AARCH64: "pytorch/manylinuxaarch64-builder:cuda12.4",
166+
CUDA_AARCH64: "pytorch/manylinuxaarch64-builder:cuda12.6",
167167
}
168168
LIBTORCH_CONTAINER_IMAGES = {
169169
**{
@@ -191,7 +191,7 @@ def translate_desired_cuda(gpu_arch_type: str, gpu_arch_version: str) -> str:
191191
return {
192192
CPU: "cpu",
193193
CPU_AARCH64: CPU,
194-
CUDA_AARCH64: "cu124",
194+
CUDA_AARCH64: "cu126",
195195
CUDA: f"cu{gpu_arch_version.replace('.', '')}",
196196
ROCM: f"rocm{gpu_arch_version}",
197197
XPU: "xpu",

0 commit comments

Comments
 (0)