From 6f3de3eb455f686f5201b3bef59b734445a0a690 Mon Sep 17 00:00:00 2001
From: Eta <esyra@coreweave.com>
Date: Mon, 8 Apr 2024 19:42:31 -0400
Subject: [PATCH] build: Label the `base` build stage

---
 Dockerfile.ubuntu20 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.ubuntu20 b/Dockerfile.ubuntu20
index 86ccf8a..bfa2a39 100644
--- a/Dockerfile.ubuntu20
+++ b/Dockerfile.ubuntu20
@@ -1,6 +1,6 @@
 ARG CUDA_VERSION_MINOR=11.7.1
 ARG BASE_IMAGE=nvidia/cuda:${CUDA_VERSION_MINOR}-cudnn8-devel-ubuntu20.04
-FROM ${BASE_IMAGE}
+FROM ${BASE_IMAGE} as base
 
 ARG CUDA_VERSION_MAJOR=11.7
 ARG TARGET_NCCL_VERSION=2.14.3-1