Blame view
tools/extras/install_tensorflow_py.sh
348 Bytes
8dcb6dfcb
first commit
|
1 2 3 4 5 6 7 8 9 10 11 12 13 |
#!/bin/bash
export HOME=$PWD/tensorflow_build/
has_gpu=true
tf_source=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp27-none-linux_x86_64.whl
if [ $has_gpu != "true" ]; then
tf_source=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.2.0-cp27-none-linux_x86_64.whl
fi
pip install --user $tf_source
|