diff --git a/requirements.txt b/requirements.txt index df7985d..da31c0d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -45,7 +45,7 @@ tensorboard==2.2.1 tensorboard-plugin-wit==1.6.0.post3 tensorflow==2.2.0 tensorflow-estimator==2.2.0 -tensorflow-gpu==2.2.0 +tensorflow-gpu==2.5.2 tensorflow-gpu-estimator==2.2.0 termcolor==1.1.0 urllib3==1.25.9