Files
ray-project--ray/ci/docker/ray-ml-anyscale-cuda.wanda.yaml
T
2026-07-13 13:17:40 +08:00

21 lines
746 B
YAML

# Ray-ML Anyscale CUDA Image (for release tests)
# Installs ray wheel into the ray-ml CUDA base-extra-testdeps image, which includes
# extra test dependencies.
#
# This produces the anyscale test image for ray-ml GPU release tests.
#
name: "ray-ml-anyscale-py$PYTHON_VERSION-$GPU$ARCH_SUFFIX"
disable_caching: true
froms:
- "cr.ray.io/rayproject/ray-ml-py$PYTHON_VERSION-$GPU-base-extra-testdeps" # ray-ml base with test deps
- "cr.ray.io/rayproject/ray-wheel-py$PYTHON_VERSION$ARCH_SUFFIX" # Ray wheel
dockerfile: ci/docker/ray-image.Dockerfile
build_args:
- PYTHON_VERSION
- ARCH_SUFFIX
- BASE_VARIANT=base-extra-testdeps
- IMAGE_TYPE=ray-ml
- PLATFORM=$GPU
- RAY_COMMIT=$BUILDKITE_COMMIT
- RAY_VERSION