build cu128 ul24

Author: Yadan Wei

base/x86_64/gpu/cu128/ubuntu24.04/Dockerfile

@@ -4,7 +4,8 @@ ARG PYTHON_SHORT_VERSION="3.12"
 ARG CUDA_MAJOR="12"
 ARG CUDA_MINOR="8"
 ARG EFA_VERSION="1.42.0"
-FROM nvidia/cuda:12.8.1-base-ubuntu24.04 AS base-builder
+ARG OS_VERSION="ubuntu24.04"
+FROM nvidia/cuda:12.8.1-base-${OS_VERSION} AS base-builder
 
 
 RUN mv /usr/local/cuda/compat /usr/local \
@@ -61,11 +62,12 @@ RUN bash install_python.sh ${PYTHON_VERSION} && rm install_python.sh
 FROM base-builder AS cuda-builder
 ARG CUDA_MAJOR
 ARG CUDA_MINOR
+ARG OS_VERSION
 COPY install_cuda.sh install_cuda.sh
-RUN bash install_cuda.sh "${CUDA_MAJOR}.${CUDA_MINOR}" && rm install_cuda.sh
+RUN bash install_cuda.sh "${CUDA_MAJOR}.${CUDA_MINOR}" "${OS_VERSION}" && rm install_cuda.sh
 
 ##############################################################################
-FROM nvidia/cuda:12.8.1-base-ubuntu24.04 AS final 
+FROM nvidia/cuda:12.8.1-base-${OS_VERSION} AS final 
 ARG PYTHON
 ARG PYTHON_SHORT_VERSION
 ARG CUDA_MAJOR

dlc_developer_config.toml

@@ -119,7 +119,7 @@ use_scheduler = false
 ### TRAINING PR JOBS ###
 
 # Base
-dlc-pr-base = "base/buildspec-cu128-ubuntu22.yml"
+dlc-pr-base = "base/buildspec-cu128-ubuntu24.yml"
 
 # Standard Framework Training
 dlc-pr-pytorch-training = ""