diff --git a/requirements.txt b/requirements.txt index 2c9acdb..22b5287 100644 --- a/requirements.txt +++ b/requirements.txt @@ -131,7 +131,7 @@ tables==3.5.2 tensorboard==1.14.0 tensorflow-datasets==1.2.0 tensorflow-estimator==1.14.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.5.2 tensorflow-metadata==0.14.0 tensorflow-probability==0.7.0 termcolor==1.1.0