ray/release/nightly_tests/dataset/ray_sgd_training_app.yaml
Chen Shen 107aef89a8
[CUJ2] add nightly tests for running 500GB ray train (#20195)
* add

* update cluster env

* fix build

Co-authored-by: Matthew Deng <matthew.j.deng@gmail.com>
2021-11-21 20:04:45 -08:00

9 lines
250 B
YAML

base_image: "anyscale/ray:nightly-py37-cu102"
python:
pip_packages: []
conda_packages: []
post_build_cmds:
- pip3 install llvmlite --ignore-installed
- pip3 install -U torch==1.6 dask dask_ml mlflow tensorboard torchvision pickle5 s3fs