driver_requirements.txt 394 B

12345678910
  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/requirements_dl.txt
  6. tensorflow
  7. torch
  8. # Need this library to unpickle errors
  9. tblib