mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05: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
13 lines
346 B
Text
13 lines
346 B
Text
ipython
|
|
|
|
# Needed for Ray Client error message serialization/deserialization.
|
|
tblib
|
|
|
|
|
|
# In TF >v2, GPU support is included in the base package.
|
|
tensorflow==2.5.0
|
|
tensorflow-probability==0.13.0
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torch==1.9.0+cu111
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
torchvision==0.10.0+cu111
|