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

freeze tensorflow version to <= 1.14.0 (#416)

Browse Source
This commit is contained in:
shadiendrawis
2019-10-10 17:47:25 +03:00
committed by GitHub
parent 5ad5a58350
commit 6db695ad8a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@@ -73,11 +73,11 @@ if not using_GPU:
# (reproduced with intel-tensorflow 1.14.0 but not with 1.13.1) # (reproduced with intel-tensorflow 1.14.0 but not with 1.13.1)
install_requires.append('intel-tensorflow==1.13.1') install_requires.append('intel-tensorflow==1.13.1')
else: else:
install_requires.append('tensorflow>=1.9.0') install_requires.append('tensorflow>=1.9.0,<=1.14.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:
install_requires.append('tensorflow-gpu>=1.9.0') install_requires.append('tensorflow-gpu>=1.9.0,<=1.14.0')
extras['mxnet'] = ['mxnet-cu90mkl>=1.3.0'] extras['mxnet'] = ['mxnet-cu90mkl>=1.3.0']
all_deps = [] all_deps = []
@@ -88,7 +88,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='1.0.0', version='1.0.1',
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',