diff --git a/setup.py b/setup.py index 524c733..9dbd567 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ from setuptools import find_packages from setuptools import setup -REQUIRED_PACKAGES = ['tensorflow-gpu==2.0.0a', 'Pillow==5.4.1', 'h5py', 'matplotlib', 'PyMaxflow'] +REQUIRED_PACKAGES = ['tensorflow-gpu==2.9.3', 'Pillow==5.4.1', 'h5py', 'matplotlib', 'PyMaxflow'] setup( name='trainer',