OPF Ray images include Ray Core and some commonly used data science libraries. GPU images add NVIDIA-specific components for accelerated workloads.
- Ray
- teradata-ray
- modin
- polars
- daft
- lightgbm-ray
- nltk
- transformers
- boto3, s3fs, smart_open
- CUDA
- cuDNN
- NVIDIA RAPIDS including cuDF and cuML
- Polars
You can also install additional Python packages at runtime when needed.
import ray
ray.init(
address="<ray-client-endpoint>",
runtime_env={"pip": ["scikit-learn==1.5.0", "xgboost"]}
)