driver_requirements.txt 354 B

12345678
  1. # Make sure the driver versions are the same as cluster versions.
  2. # The cluster uses ray-ml Docker image.
  3. # ray-ml Docker image installs dependencies from ray/python/requirements/ml/ directory.
  4. # We constrain on these requirements file so that the same versions are installed.
  5. -c ../../../python/requirements/ml/dl-cpu-requirements.txt
  6. torch
  7. torchvision