2021-11-21 20:04:45 -08:00
|
|
|
base_image: "anyscale/ray:nightly-py37-cu102"
|
|
|
|
|
|
|
|
|
|
python:
|
2022-02-10 14:22:41 -08:00
|
|
|
pip_packages:
|
|
|
|
|
- pyarrow<7.0.0
|
2021-11-21 20:04:45 -08:00
|
|
|
conda_packages: []
|
|
|
|
|
|
|
|
|
|
post_build_cmds:
|
2022-05-22 16:07:03 +09:00
|
|
|
- pip uninstall -y ray && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
|
2021-11-21 20:04:45 -08:00
|
|
|
- pip3 install llvmlite --ignore-installed
|
|
|
|
|
- pip3 install -U torch==1.6 dask dask_ml mlflow tensorboard torchvision pickle5 s3fs
|
2021-12-17 23:48:44 -08:00
|
|
|
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
|
