Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Adjust docker gpu aarch64 docker image #1794

Merged
merged 3 commits into from
Apr 23, 2024
Merged

Conversation

atalman
Copy link
Contributor

@atalman atalman commented Apr 23, 2024

Follow up after #1775

This should fix this error: https://github.com/pytorch/builder/actions/runs/8792217019/job/24127939462#step:4:15

 docker build -t docker.io/pytorch/manylinuxaarch64-builder:cuda-aarch64 --build-arg BASE_CUDA_VERSION=12.4 --build-arg DEVTOOLSET_VERSION=11 --build-arg GPU_IMAGE=arm64v8/centos:7 --target cuda_final -f /home/ec2-user/actions-runner/_work/builder/builder/manywheel/Dockerfile_aarch64 /home/ec2-user/actions-runner/_work/builder/builder
#1 [internal] load build definition from Dockerfile_aarch64

Since Dockerfile_cuda_aarch64 has to be used. This keeps same docker image:
From:

docker.io/pytorch/manylinuxaarch64-builder:cuda-aarch64

@atalman atalman merged commit 8fd2559 into pytorch:main Apr 23, 2024
@atalman atalman deleted the fix_cuda_aarch64 branch April 23, 2024 19:02
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants