@@ -22,6 +22,11 @@ while [[ $# -gt 0 ]]; do
22
22
shift
23
23
BASE_IMAGE=" $1 "
24
24
;;
25
+ --progress-plain)
26
+ # Use plain progress output instead of fancy output.
27
+ # This is useful for CI systems that don't support fancy output.
28
+ BUILD_ARGS+=(" --progress=plain" )
29
+ ;;
25
30
--no-cache-build)
26
31
BUILD_ARGS+=(" --no-cache" )
27
32
;;
@@ -40,7 +45,7 @@ while [[ $# -gt 0 ]]; do
40
45
PYTHON_VERSION=" $1 "
41
46
;;
42
47
* )
43
- echo " Usage: build-docker.sh [ --gpu ] [ --base-image ] [ --no-cache-build ] [ --shas-only ] [ --build-development-image ] [ --build-examples ] [ --python-version ]"
48
+ echo " Usage: build-docker.sh [ --gpu ] [ --base-image ] [ --no-cache-build ] [ --shas-only ] [ --progress-plain ] [ --python-version ]"
44
49
exit 1
45
50
esac
46
51
shift
@@ -54,11 +59,17 @@ if [[ "$OUTPUT_SHA" != "YES" ]]; then
54
59
echo " === Building base-deps image ===" > /dev/stderr
55
60
fi
56
61
62
+ RAY_DEPS_BUILD_DIR=" $( mktemp -d) "
63
+
64
+ cp docker/base-deps/Dockerfile " ${RAY_DEPS_BUILD_DIR} /."
65
+ mkdir -p " ${RAY_DEPS_BUILD_DIR} /python"
66
+ cp python/requirements_compiled.txt " ${RAY_DEPS_BUILD_DIR} /python/requirements_compiled.txt"
67
+
57
68
BUILD_CMD=(
58
69
docker build " ${BUILD_ARGS[@]} "
59
70
--build-arg BASE_IMAG=" $BASE_IMAGE "
60
71
--build-arg PYTHON_VERSION=" ${PYTHON_VERSION} "
61
- -t " rayproject/base-deps:dev$GPU " " docker/base-deps "
72
+ -t " rayproject/base-deps:dev$GPU " " ${RAY_DEPS_BUILD_DIR} "
62
73
)
63
74
64
75
if [[ " $OUTPUT_SHA " == " YES" ]]; then
@@ -78,7 +89,6 @@ RAY_BUILD_DIR="$(mktemp -d)"
78
89
mkdir -p " $RAY_BUILD_DIR /.whl"
79
90
wget --quiet " $WHEEL_URL " -P " $RAY_BUILD_DIR /.whl"
80
91
wget --quiet " $CPP_WHEEL_URL " -P " $RAY_BUILD_DIR /.whl"
81
- cp python/requirements_compiled.txt " $RAY_BUILD_DIR "
82
92
cp docker/ray/Dockerfile " $RAY_BUILD_DIR "
83
93
84
94
WHEEL=" $( basename " $WHEEL_DIR " /.whl/ray-* .whl) "
@@ -87,7 +97,7 @@ BUILD_CMD=(
87
97
docker build " ${BUILD_ARGS[@]} "
88
98
--build-arg FULL_BASE_IMAGE=" rayproject/base-deps:dev$GPU "
89
99
--build-arg WHEEL_PATH=" .whl/${WHEEL} "
90
- -t " rayproject/ray:dev$GPU " " $RAY_BUILD_DIR "
100
+ -t " rayproject/ray:dev$GPU " " ${ RAY_BUILD_DIR} "
91
101
)
92
102
93
103
if [[ " $OUTPUT_SHA " == " YES" ]]; then
0 commit comments