[Docker] usage of python-version (#13011)

Co-authored-by: Tanja Bayer <tanja.bayer@widas.de>
Co-authored-by: Ian Rodney <ian.rodney@gmail.com>
This commit is contained in:
Tanja Bayer
2021-01-28 23:27:54 +01:00
committed by GitHub
parent cb771f263d
commit 0c906a8b93
+4 -2
View File
@@ -8,7 +8,8 @@ set -x
GPU="" GPU=""
BASE_IMAGE="ubuntu:focal" BASE_IMAGE="ubuntu:focal"
WHEEL_URL="https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp37-cp37m-manylinux2014_x86_64.whl" WHEEL_URL="https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp37-cp37m-manylinux2014_x86_64.whl"
PYTHON_VERSION="" PYTHON_VERSION="3.7.7"
while [[ $# -gt 0 ]] while [[ $# -gt 0 ]]
do do
@@ -41,6 +42,7 @@ case $key in
--python-version) --python-version)
# Python version to install. e.g. 3.7.7. # Python version to install. e.g. 3.7.7.
# Changing python versions may require a different wheel. # Changing python versions may require a different wheel.
# If not provided defaults to 3.7.7
shift shift
PYTHON_VERSION=$1 PYTHON_VERSION=$1
;; ;;
@@ -59,7 +61,7 @@ for IMAGE in "base-deps" "ray-deps" "ray"
do do
cp "$WHEEL" "docker/$IMAGE/$(basename "$WHEEL")" cp "$WHEEL" "docker/$IMAGE/$(basename "$WHEEL")"
if [ $OUTPUT_SHA ]; then 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:nightly$GPU docker/$IMAGE) IMAGE_SHA=$(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" -q -t rayproject/$IMAGE:nightly$GPU docker/$IMAGE)
echo "rayproject/$IMAGE:nightly$GPU SHA:$IMAGE_SHA" echo "rayproject/$IMAGE:nightly$GPU SHA:$IMAGE_SHA"
else 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:nightly$GPU 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:nightly$GPU docker/$IMAGE