From fdf4bb50871c01bc90ab0530bcceb5d79e9f8eea Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Fri, 6 Dec 2024 11:35:40 -0500
Subject: [PATCH] fix default base image

---
 docker/Dockerfile-base | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
index ab118cef9..9f65d265a 100644
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -6,7 +6,7 @@ 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:-${DEFAULT_TAG}} AS base-builder
+FROM ${BASE_IMAGE:-nvidia/cuda}:${BASE_TAG:-${DEFAULT_TAG}} AS base-builder
 
 ENV PATH="/root/miniconda3/bin:${PATH}"