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

bug fix in dueling network + revert to TF 1.6 for CPU due to requirements compatibility issues

Browse Source
This commit is contained in:
itaicaspi-intel
2018-09-02 13:38:16 +03:00
parent 3a399d1361
commit 2c62a40466
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -61,9 +61,9 @@ if not using_GPU:
# 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'],
'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.10.0')
install_requires.append('tensorflow==1.6.0')
else:
install_requires.append('tensorflow-gpu==1.10.0')