diff --git a/requirements.txt b/requirements.txt index a0aa49d..ebf0108 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,7 +33,7 @@ six==1.15.0 sklearn==0.0 tensorboard==1.13.1 tensorflow-estimator==1.13.0 -tensorflow-gpu==1.13.1 +tensorflow-gpu==2.12.0 tensorflow-probability==0.6.0 termcolor==1.1.0 threadpoolctl==2.1.0