build: Update Dockerfile build argument defaults

coreweave · Oct 31, 2024 · 66b413d · 66b413d
1 parent ccaa59d
commit 66b413d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/Dockerfile.ubuntu20 b/Dockerfile.ubuntu20
@@ -1,9 +1,9 @@
-ARG CUDA_VERSION_MINOR=11.7.1
-ARG BASE_IMAGE=nvidia/cuda:${CUDA_VERSION_MINOR}-cudnn8-devel-ubuntu20.04
+ARG CUDA_VERSION_MINOR=12.4.1
+ARG BASE_IMAGE=nvidia/cuda:${CUDA_VERSION_MINOR}-cudnn-devel-ubuntu20.04
 FROM ${BASE_IMAGE} as base
 
-ARG CUDA_VERSION_MAJOR=11.7
-ARG TARGET_NCCL_VERSION=2.14.3-1
+ARG CUDA_VERSION_MAJOR=12.4
+ARG TARGET_NCCL_VERSION=2.23.4-1
 
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get -qq update && \

diff --git a/Dockerfile.ubuntu22 b/Dockerfile.ubuntu22
@@ -1,9 +1,9 @@
-ARG CUDA_VERSION_MINOR=12.2.2
-ARG BASE_IMAGE=nvidia/cuda:${CUDA_VERSION_MINOR}-cudnn8-devel-ubuntu22.04
+ARG CUDA_VERSION_MINOR=12.4.1
+ARG BASE_IMAGE=nvidia/cuda:${CUDA_VERSION_MINOR}-cudnn-devel-ubuntu22.04
 FROM ${BASE_IMAGE} as base
 
-ARG CUDA_VERSION_MAJOR=12.2
-ARG TARGET_NCCL_VERSION=2.19.3-1
+ARG CUDA_VERSION_MAJOR=12.4
+ARG TARGET_NCCL_VERSION=2.23.4-1
 
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get -qq update && \