mirror of
https://github.com/vale981/ray
synced 2025-03-10 13:26:39 -04:00
3a52187da8
* wip * wip * add ray lightning test * fix * update * merge and add * fix * fix * rename * autoscale * add tblib * gloo backend * typo * upgrade torch * latest and master
9 lines
No EOL
365 B
Text
9 lines
No EOL
365 B
Text
# 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
|
|
pytorch-lightning |