ray/release/nightly_tests/dataset/ray_sgd_training_app.yaml

10 lines
250 B
YAML
Raw Normal View History

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