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