diff --git a/requirements.txt b/requirements.txt index e358c01..6752966 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ PyMySQL>=0.9.3 pandas>=0.19.2 numpy>=1.14.5 tensorflow==1.15 -tensorflow-gpu==1.15 +tensorflow-gpu==2.9.3 keras==2.2.4 scikit-learn==0.22 pynvml==8.0.3