diff --git a/requirements.txt b/requirements.txt index dbb129c..e477377 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -tensorflow-gpu == 1.15.2 +tensorflow-gpu == 2.12.0 requests == 2.23.0 numpy == 1.18.2 matplotlib == 3.2.1 diff --git a/setup.py b/setup.py index 89b3825..3a39568 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ def run(self): 'License :: OSI Approved :: MIT License', 'Operating System :: OS Independent'], install_requires=[ - 'tensorflow-gpu==1.15.2', + 'tensorflow-gpu==2.12.0', 'requests==2.23.0', 'numpy==1.18.2', 'matplotlib==3.2.1',