From 3a399d1361ad54fda10066081e2a98d58c10c717 Mon Sep 17 00:00:00 2001
From: Itai Caspi <30383381+itaicaspi-intel@users.noreply.github.com>
Date: Sun, 2 Sep 2018 10:12:00 +0300
Subject: [PATCH] Tensorflow 1.10 and python 3.6 (#104)

* updating setup.py to install tensorflow 1.10 both on cpu and on gpu

* allow python 3.6
---
 setup.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index 018259b..bd779d2 100644
--- a/setup.py
+++ b/setup.py
@@ -58,12 +58,14 @@ out = p.communicate()[0].decode('UTF-8')
 using_GPU = out != ''
 
 if not using_GPU:
-    subprocess.check_call(['pip install '
-                           'https://anaconda.org/intel/tensorflow/1.6.0/download/tensorflow-1.6.0-cp35-cp35m-linux_x86_64.whl'],
-                          shell=True)
-    install_requires.append('tensorflow==1.6.0')
+    # For linux wth no GPU, we install the Intel optimized version of TensorFlow
+    if sys.platform == "linux" or sys.platform == "linux2":
+        subprocess.check_call(['pip install '
+                               'https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.10.0-cp35-cp35m-linux_x86_64.whl'],
+                              shell=True)
+    install_requires.append('tensorflow==1.10.0')
 else:
-    install_requires.append('tensorflow-gpu==1.9.0')
+    install_requires.append('tensorflow-gpu==1.10.0')
 
 setup(
     name='rl-coach',
@@ -73,7 +75,7 @@ setup(
     author='Intel AI Lab',
     author_email='coach@intel.com',
     packages=find_packages(),
-    python_requires="==3.5.*",
+    python_requires=">=3.5.*",
     install_requires=install_requires,
     package_data={'rl_coach': ['dashboard_components/*.css',
                                'environments/doom/*.cfg',