gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

fix Intel tensorflow installation issue (#281)

Browse Source 
* fix intel tensorflow installation issue

* update version
This commit is contained in:
shadiendrawis
2019-04-03 13:03:30 +03:00
committed by GitHub
parent 5d4b9c7399
commit a543f10c1a
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
setup.py
View File

@@ -68,10 +68,9 @@ if not using_GPU:
if not slim_package: if not slim_package:
# For linux wth no GPU, we install the Intel optimized version of TensorFlow # For linux wth no GPU, we install the Intel optimized version of TensorFlow
if sys.platform == "linux" or sys.platform == "linux2": if sys.platform == "linux" or sys.platform == "linux2":
subprocess.check_call(['pip install ' install_requires.append('intel-tensorflow>=1.9.0')
'https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.12.0-cp35-cp35m-linux_x86_64.whl'], else:
shell=True) install_requires.append('tensorflow>=1.9.0')
install_requires.append('tensorflow>=1.9.0')
extras['mxnet'] = ['mxnet-mkl>=1.3.0'] extras['mxnet'] = ['mxnet-mkl>=1.3.0']
else: else:
if not slim_package: if not slim_package:
@@ -86,7 +85,7 @@ extras['all'] = all_deps
setup( setup(
name='rl-coach' if not slim_package else 'rl-coach-slim', name='rl-coach' if not slim_package else 'rl-coach-slim',
version='0.11.1', version='0.11.2',
description='Reinforcement Learning Coach enables easy experimentation with state of the art Reinforcement Learning algorithms.', description='Reinforcement Learning Coach enables easy experimentation with state of the art Reinforcement Learning algorithms.',
url='https://github.com/NervanaSystems/coach', url='https://github.com/NervanaSystems/coach',
author='Intel AI Lab', author='Intel AI Lab',