Skip to content
This repository has been archived by the owner on Dec 11, 2022. It is now read-only.

Commit

Permalink
fix Intel tensorflow installation issue (#281)
Browse files Browse the repository at this point in the history
* fix intel tensorflow installation issue

* update version
  • Loading branch information
shadiendrawis authored Apr 3, 2019
1 parent 5d4b9c7 commit a543f10
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,10 +68,9 @@
if not slim_package:
# 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.12.0-cp35-cp35m-linux_x86_64.whl'],
shell=True)
install_requires.append('tensorflow>=1.9.0')
install_requires.append('intel-tensorflow>=1.9.0')
else:
install_requires.append('tensorflow>=1.9.0')
extras['mxnet'] = ['mxnet-mkl>=1.3.0']
else:
if not slim_package:
Expand All @@ -86,7 +85,7 @@

setup(
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.',
url='https://github.com/NervanaSystems/coach',
author='Intel AI Lab',
Expand Down

0 comments on commit a543f10

Please sign in to comment.