2021-07-16 14:15:49 -07:00
|
|
|
base_image: "anyscale/ray-ml:pinned-nightly-py37-gpu"
|
2021-09-28 01:29:26 -07:00
|
|
|
env_vars: {"RAY_event_log_reporter_enabled": "1"}
|
2021-07-16 14:15:49 -07:00
|
|
|
|
|
|
|
python:
|
|
|
|
pip_packages: []
|
|
|
|
conda_packages: []
|
|
|
|
|
|
|
|
post_build_cmds:
|
|
|
|
- pip uninstall -y numpy ray || true
|
|
|
|
- sudo rm -rf /home/ray/anaconda3/lib/python3.7/site-packages/numpy
|
|
|
|
- pip3 install boto3 pyarrow tqdm
|
|
|
|
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
|
2021-09-10 17:50:31 +01:00
|
|
|
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
|