diff --git a/docker/Dockerfile.build.aarch64-linux b/docker/Dockerfile.build.aarch64-linux index 6d7976cd2a..3b50a6fbe3 100644 --- a/docker/Dockerfile.build.aarch64-linux +++ b/docker/Dockerfile.build.aarch64-linux @@ -1,9 +1,9 @@ -ARG AARCH64_BASE_IMAGE=nvidia/cuda:12.2.0-devel-ubuntu20.04 +ARG AARCH64_BASE_IMAGE=nvidia/cuda:12.6.3-devel-ubuntu22.04 FROM ${AARCH64_BASE_IMAGE} ENV DEBIAN_FRONTEND=noninteractive \ - CUDA_CROSS_VERSION=12-2 \ - CUDA_CROSS_VERSION_DOT=12.2 + CUDA_CROSS_VERSION=12-6 \ + CUDA_CROSS_VERSION_DOT=12.6 RUN rm /etc/apt/sources.list.d/cuda.list && \ apt-key del 7fa2af80 && \