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