Update setup.py

AICONSlab · Oct 27, 2021 · 177b459 · 177b459
1 parent 79aef31
commit 177b459
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -40,8 +40,8 @@
         'keras-contrib', 'pandas', 'numpy', 'plotly', 'PyQt5', 'termcolor'
     ],
     extras_require={
-        "icvmapper": ["tensorflow"],
-        "icvmapper_gpu": ["tensorflow-gpu"],
+        "icvmapper": ["tensorflow==1.15"],
+        "icvmapper_gpu": ["tensorflow-gpu==1.15"],
     },
     entry_points={'console_scripts': ['icvmapper=icvmapper.cli:main']},
     keywords=[