absl-py ~= 0.10
h5py ~= 2.10.0
keras_preprocessing ~= 1.1.2
numpy ~= 1.19.2
pandas
pydot
scipy ~= 1.5.2
six ~= 1.15.0
tf-nightly-gpu == 2.5.0.dev20210129
portpicker
pyyaml
Pillow

absl-py ~= 0.10
h5py ~= 2.10.0
keras_preprocessing ~= 1.1.2
numpy ~= 1.19.2
pandas
pydot
scipy ~= 1.5.2
six ~= 1.15.0
tf-nightly-gpu
portpicker
pyyaml
Pillow

Use the latest tf-nightly since the OOM issue from nightly might be fixed.
Use the latest tf-nightly since the OOM issue from nightly might be fixed.

PiperOrigin-RevId: 355451756
