mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
13 lines
445 B
YAML
13 lines
445 B
YAML
base_image: "anyscale/ray:nightly-py37-cu102"
|
|
|
|
python:
|
|
pip_packages:
|
|
- pyarrow<7.0.0
|
|
conda_packages: []
|
|
|
|
post_build_cmds:
|
|
- pip uninstall -y ray
|
|
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
|
|
- pip3 install llvmlite --ignore-installed
|
|
- pip3 install -U torch==1.6 dask dask_ml mlflow tensorboard torchvision pickle5 s3fs
|
|
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
|