Issue yell#8 Use tensorflow < 2.0.0

SrMouraSilva · Mar 6, 2019 · c6a96a4 · c6a96a4
1 parent ade4eaf
commit c6a96a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -4,13 +4,13 @@
 setup(name='boltzmann_machines',
       version='0.1',
       url='https://github.com/monsta-hd/boltzmann-machines',
-      install_requires=['tensorflow-gpu>=1.3.0',
+      install_requires=['tensorflow==1.13.*',
                         'scipy>=0.17', 
                         'keras>=2.0.8', 
                         'matplotlib>=1.5.3', 
                         'nose-exclude>=0.5.0', 
                         'nose>=1.3.4', 
-                        'numpy>=1.13.0',
+                        'numpy>=1.16.0',
                         'scikit-learn>=0.19.0',
                         'seaborn>=0.8.1', 
                         'tqdm>=4.14.0',

diff --git a/setup_cpu.py b/setup_cpu.py
@@ -4,13 +4,13 @@
 setup(name='boltzmann_machines',
       version='0.1',
       url='https://github.com/monsta-hd/boltzmann-machines',
-      install_requires=['tensorflow>=1.13.0',
+      install_requires=['tensorflow==1.13.*',
                         'scipy>=0.17', 
                         'keras>=2.0.8', 
                         'matplotlib>=1.5.3', 
                         'nose-exclude>=0.5.0', 
                         'nose>=1.3.4', 
-                        'numpy>=1.13.0',
+                        'numpy>=1.16.0',
                         'scikit-learn>=0.19.0',
                         'seaborn>=0.8.1', 
                         'tqdm>=4.14.0',