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:
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
|
||||
else
|
||||
# GPU supported TensorFlow
|
||||
pip3 install tensorflow-gpu
|
||||
pip3 install tensorflow-gpu==1.4.1
|
||||
fi
|
||||
|
||||
Reference in New Issue
Block a user