mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 15:16:34 +08:00
[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:
+4
-2
@@ -8,7 +8,8 @@ set -x
|
||||
GPU=""
|
||||
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"
|
||||
PYTHON_VERSION=""
|
||||
PYTHON_VERSION="3.7.7"
|
||||
|
||||
|
||||
while [[ $# -gt 0 ]]
|
||||
do
|
||||
@@ -41,6 +42,7 @@ case $key in
|
||||
--python-version)
|
||||
# Python version to install. e.g. 3.7.7.
|
||||
# Changing python versions may require a different wheel.
|
||||
# If not provided defaults to 3.7.7
|
||||
shift
|
||||
PYTHON_VERSION=$1
|
||||
;;
|
||||
@@ -59,7 +61,7 @@ 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: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"
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user