try with default tag

2024-12-06 11:10:49 -05:00
parent 684b543aa1
commit f67d16268c
1 changed files with 2 additions and 1 deletions
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -4,8 +4,9 @@ ARG UBUNTU_VERSION="22.04"
 ARG MAX_JOBS=4

 ARG BASE_IMAGE=nvidia/cuda
+ARG DEFAULT_TAG=${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}
 ARG BASE_TAG=""
-FROM $BASE_IMAGE:${BASE_TAG:-$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_VERSION} AS base-builder
+FROM ${BASE_IMAGE}:${BASE_TAG:-${DEFAULT_TAG}} AS base-builder

 ENV PATH="/root/miniconda3/bin:${PATH}"