Move to JP6.1

Signed-off-by: Joaquin Anton Guirao <[email protected]>
NVIDIA · Jan 16, 2025 · 242aff4 · 242aff4
1 parent 65f279d
commit 242aff4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 && \