@@ -67,10 +67,11 @@ function install_protobuf3.4.0() {
67
67
# Utility function for installing tensorflow components of python/C++
68
68
function install_tf() {
69
69
TFCApiFile=$1
70
- LinkerConfigCmd=$2
70
+ TFBinaryURL=$2
71
+ LinkerConfigCmd=$3
71
72
TARGET_DIRECTORY=" /usr/local"
72
73
# Install Tensorflow Python Binary
73
- sudo -E pip3 install tensorflow== ${TF_VERSION }
74
+ sudo -E pip3 install --upgrade ${TFBinaryURL }
74
75
75
76
# Install C-API
76
77
TFCApiURL=" https://storage.googleapis.com/tensorflow/libtensorflow/${TFCApiFile} "
@@ -171,7 +172,7 @@ if [ "$DISTRO" = "UBUNTU" ]; then
171
172
# Install version of protobuf needed by C-API
172
173
install_protobuf3.4.0 " ubuntu"
173
174
# Install tensorflow
174
- install_tf " $TFCApiFile " " $LinkerConfigCmd "
175
+ install_tf " $TFCApiFile " " $TFBinaryURL " " $ LinkerConfigCmd"
175
176
176
177
# # ------------------------------------------------
177
178
# # DEBIAN
@@ -247,7 +248,7 @@ elif [[ "$DISTRO" == *"FEDORA"* ]]; then
247
248
# Install version of protobuf needed by C-API
248
249
install_protobuf3.4.0 " fedora"
249
250
# Install tensorflow
250
- install_tf " $TFCApiFile " " $LinkerConfigCmd "
251
+ install_tf " $TFCApiFile " " $TFBinaryURL " " $ LinkerConfigCmd"
251
252
252
253
# # ------------------------------------------------
253
254
# # REDHAT
@@ -352,7 +353,7 @@ elif [ "$DISTRO" = "DARWIN" ]; then
352
353
brew upgrade python
353
354
# Brew installs correct version of Protobuf(3.5.1 >= 3.4.0)
354
355
# So we can directly install tensorflow
355
- install_tf " $TFCApiFile " " $LinkerConfigCmd "
356
+ install_tf " $TFCApiFile " " $TFBinaryURL " " $ LinkerConfigCmd"
356
357
357
358
# # ------------------------------------------------
358
359
# # UNKNOWN
0 commit comments