Files
2026-07-13 13:17:40 +08:00

9 lines
354 B
Plaintext
Executable File

# Make sure the driver versions are the same as cluster versions.
# The cluster uses ray-ml Docker image.
# ray-ml Docker image installs dependencies from ray/python/requirements/ml/ directory.
# We constrain on these requirements file so that the same versions are installed.
-c ../../../python/requirements/ml/dl-cpu-requirements.txt
torch
tensorflow