diff --git a/requirements.txt b/requirements.txt index b4c2a07..cbcb6e5 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.12.0 tensorflow-metadata==0.13.0 tensorflow-probability==0.7.0