From 1deb281ea6a33b169fda90e8910c002f2ba0e48e Mon Sep 17 00:00:00 2001 From: Lee moon soo Date: Tue, 22 Sep 2020 10:19:53 -0700 Subject: [PATCH] [docker] Add $GPU to image tag name (#10909) --- build-docker.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/build-docker.sh b/build-docker.sh index f1d59bc3d..0f9a3f19e 100755 --- a/build-docker.sh +++ b/build-docker.sh @@ -59,10 +59,10 @@ for IMAGE in "base-deps" "ray-deps" "ray" do cp "$WHEEL" "docker/$IMAGE/$(basename "$WHEEL")" if [ $OUTPUT_SHA ]; then - IMAGE_SHA=$(docker build $NO_CACHE --build-arg GPU="$GPU" --build-arg BASE_IMAGE="$BASE_IMAGE" --build-arg WHEEL_PATH="$(basename "$WHEEL")" -q -t rayproject/$IMAGE:latest docker/$IMAGE) - echo "rayproject/$IMAGE:latest SHA:$IMAGE_SHA" + IMAGE_SHA=$(docker build $NO_CACHE --build-arg GPU="$GPU" --build-arg BASE_IMAGE="$BASE_IMAGE" --build-arg WHEEL_PATH="$(basename "$WHEEL")" -q -t rayproject/$IMAGE:latest$GPU docker/$IMAGE) + echo "rayproject/$IMAGE:latest$GPU SHA:$IMAGE_SHA" else - docker build $NO_CACHE --build-arg GPU="$GPU" --build-arg BASE_IMAGE="$BASE_IMAGE" --build-arg WHEEL_PATH="$(basename "$WHEEL")" --build-arg PYTHON_VERSION="$PYTHON_VERSION" -t rayproject/$IMAGE:latest docker/$IMAGE + docker build $NO_CACHE --build-arg GPU="$GPU" --build-arg BASE_IMAGE="$BASE_IMAGE" --build-arg WHEEL_PATH="$(basename "$WHEEL")" --build-arg PYTHON_VERSION="$PYTHON_VERSION" -t rayproject/$IMAGE:latest$GPU docker/$IMAGE fi rm "docker/$IMAGE/$(basename "$WHEEL")" done