diff --git a/requirements.txt b/requirements.txt index f739f8e..32663d4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -194,7 +194,7 @@ tensorboardX==2.0 tensorflow==2.2.0 tensorflow-datasets==2.1.0 tensorflow-estimator==2.2.0 -tensorflow-gpu==2.1.0 +tensorflow-gpu==2.5.1 tensorflow-hub==0.7.0 tensorflow-metadata==0.21.1 termcolor==1.1.0