1
0
mirror of https://github.com/gryf/coach.git synced 2026-02-01 05:25:44 +01:00

Setting tensorflow-gpu version to 1.4.1 (1.5.0 is not tested yet)

This commit is contained in:
galleibo-intel
2018-02-05 15:47:06 +02:00
committed by Gal Leibovich
parent 7c8962c991
commit 4025496783

View File

@@ -200,5 +200,5 @@ if ! [ -x "$(command -v nvidia-smi)" ]; then
pip3 install https://anaconda.org/intel/tensorflow/1.3.0/download/tensorflow-1.3.0-cp35-cp35m-linux_x86_64.whl
else
# GPU supported TensorFlow
pip3 install tensorflow-gpu
pip3 install tensorflow-gpu==1.4.1
fi