mirror of
https://github.com/gryf/coach.git
synced 2026-03-18 15:53:35 +01:00
Setting tensorflow-gpu version to 1.4.1 (1.5.0 is not tested yet)
This commit is contained in:
committed by
Gal Leibovich
parent
7c8962c991
commit
4025496783
@@ -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
|
pip3 install https://anaconda.org/intel/tensorflow/1.3.0/download/tensorflow-1.3.0-cp35-cp35m-linux_x86_64.whl
|
||||||
else
|
else
|
||||||
# GPU supported TensorFlow
|
# GPU supported TensorFlow
|
||||||
pip3 install tensorflow-gpu
|
pip3 install tensorflow-gpu==1.4.1
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user