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

broaden supported package versions (#50)

Browse Source 
* broaden supported package versions.
* fix mxnet variants.
Also back-out tuple deprecation change introduced in prior commit.
* correct CI image deployment on master branch merge.
This commit is contained in:
Scott Leishman
2018-11-15 05:29:49 -08:00
committed by Gal Novik
parent 6d40ad1650
commit fe6857eabd
3 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
setup.py
View File

@@ -63,13 +63,13 @@ 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://anaconda.org/intel/tensorflow/1.6.0/download/tensorflow-1.6.0-cp35-cp35m-linux_x86_64.whl'],
'https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.11.0-cp35-cp35m-linux_x86_64.whl'],
shell=True)
install_requires.append('tensorflow==1.6.0')
extras['mxnet'] = ['mxnet-cu90mkl>=1.3.0']
else:
install_requires.append('tensorflow-gpu==1.9.0')
install_requires.append('tensorflow>=1.9.0')
extras['mxnet'] = ['mxnet-mkl>=1.3.0']
else:
install_requires.append('tensorflow-gpu>=1.9.0')
extras['mxnet'] = ['mxnet-cu90mkl>=1.3.0']
all_deps = []
for group_name in extras: