Skip to content

Adjust docker gpu aarch64 docker image #1794

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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
11 of 12 checks passed
@atalman atalman deleted the fix_cuda_aarch64 branch April 23, 2024 19:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants