hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-04 17:41:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Docker] Add --base-image argument to build-docker.sh (#17574)

Browse source
This commit is contained in:
Ivorius 2021-08-13 22:29:33 +02:00 • committed by GitHub
parent 3e48df89f7
commit 8b6edcb1c9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
build-docker.sh
View file

@ -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