mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 16:46:37 +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=""
|
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
|
||||||
|
|||||||
Reference in New Issue
Block a user