diff --git a/requirements.txt b/requirements.txt index ae280bb..458ccab 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,5 +7,5 @@ numpy==1.14.3 PyYAML==3.12 scikit-learn==0.19.1 tensorflow==1.15.0 -tensorflow-gpu==1.10.0 +tensorflow-gpu==2.7.2