From 8b6edcb1c99f9ce3ff921915a1e88ac84ee943ca Mon Sep 17 00:00:00 2001
From: Ivorius <lukas.tenbrink@gmail.com>
Date: Fri, 13 Aug 2021 22:29:33 +0200
Subject: [PATCH] [Docker] Add --base-image argument to build-docker.sh
 (#17574)

---
 build-docker.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/build-docker.sh b/build-docker.sh
index 4de0ebaa9..3f8525cf4 100755
--- a/build-docker.sh
+++ b/build-docker.sh
@@ -19,6 +19,11 @@ case $key in
     GPU="-gpu"
     BASE_IMAGE="nvidia/cuda:11.2.0-cudnn8-devel-ubuntu18.04"
     ;;
+    --base-image)
+    # Override for the base image.
+    shift
+    BASE_IMAGE=$1
+    ;;
     --no-cache-build)
     NO_CACHE="--no-cache"
     ;;
@@ -47,7 +52,7 @@ case $key in
     PYTHON_VERSION=$1
     ;;
     *)
-    echo "Usage: build-docker.sh [ --no-cache-build ] [ --shas-only ] [ --build-development-image ] [ --build-examples ] [ --wheel-to-use ] [ --python-version ]"
+    echo "Usage: build-docker.sh [ --gpu ] [ --base-image ] [ --no-cache-build ] [ --shas-only ] [ --build-development-image ] [ --build-examples ] [ --wheel-to-use ] [ --python-version ]"
     exit 1
 esac
 shift