2020-09-17 10:56:58 -07:00
|
|
|
ipython
|
2021-11-01 18:29:48 -07:00
|
|
|
|
|
|
|
|
# Needed for Ray Client error message serialization/deserialization.
|
|
|
|
|
tblib
|
|
|
|
|
|
2021-11-05 00:58:34 -07:00
|
|
|
# If you make changes to anything below this line, please also change `requirements_dl.txt`!
|
|
|
|
|
|
|
|
|
|
# TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
|
|
|
|
|
keras==2.6.0
|
2021-08-13 11:39:06 -07:00
|
|
|
# In TF >v2, GPU support is included in the base package.
|
2021-11-05 00:58:34 -07:00
|
|
|
tensorflow==2.6.0
|
|
|
|
|
tensorflow-probability==0.14.0
|
|
|
|
|
|
2021-01-27 13:45:30 -08:00
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
2021-09-16 20:16:40 -07:00
|
|
|
torch==1.9.0+cu111
|
2021-01-27 13:45:30 -08:00
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
2021-09-16 20:16:40 -07:00
|
|
|
torchvision==0.10.0+cu111
|
