diff --git a/requirements.txt b/requirements.txt index b4c2a07..a9c4fc7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,6 +5,6 @@ scikit-learn==0.21.2 tensorboard==1.14.0 tensorflow-datasets==1.0.2 tensorflow-estimator==1.14.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.5.3 tensorflow-metadata==0.13.0 tensorflow-probability==0.7.0