ray/release/nightly_tests/dataset/ray_sgd_training_app.yaml
Chen Shen c9c3f0745a
[Dataset][nighlytest] use latest ray for running test #21148
We are actually using the ray comes with the image, which is on a very old version of Ray. (suprised this actually works)
2021-12-17 23:48:44 -08:00

12 lines
428 B
YAML

base_image: "anyscale/ray:nightly-py37-cu102"
python:
pip_packages: []
conda_packages: []
post_build_cmds:
- pip uninstall -y ray
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
- pip3 install llvmlite --ignore-installed
- pip3 install -U torch==1.6 dask dask_ml mlflow tensorboard torchvision pickle5 s3fs
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}