diff --git a/setup.py b/setup.py index 3bc3de4..f8f086f 100644 --- 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=[