ray/release/ml_user_tests/horovod/driver_requirements.txt

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

8 lines
350 B
Text
Raw Permalink Normal View History

# 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/requirements_dl.txt
torch
torchvision