Trust storage.googleapis.com to avoid SSL verification problems.
@@ -27,27 +27,25 @@ rm -rf ${TENSORFLOW_LIB_DIR}
######################################################################################
echo ""
echo "Downloading and installing TensorFlow prebuilt binaries ..."
-
-
-cd ${INSTALL_DIR}
+# cd ${INSTALL_DIR}
if [[ $PYTHON3 == 0 ]]
then
if [[ $GPU_ENABLED == 1 ]]
then
- export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${TENSORFLOW_PACKAGE_VER}-cp27-none-linux_x86_64.whl
+ export TENSORFLOW_BIN_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${TENSORFLOW_PACKAGE_VER}-cp27-none-linux_x86_64.whl
else
- export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-${TENSORFLOW_PACKAGE_VER}-cp27-none-linux_x86_64.whl
+ export TENSORFLOW_BIN_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-${TENSORFLOW_PACKAGE_VER}-cp27-none-linux_x86_64.whl
fi
- pip install --upgrade $TF_BINARY_URL -t ${TENSORFLOW_LIB_DIR}
+ pip install --upgrade ${TENSORFLOW_BIN_URL} -t ${TENSORFLOW_LIB_DIR} --trusted-host storage.googleapis.com
else
if [[ $GPU_ENABLED == 1 ]]
then
- export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${TENSORFLOW_PACKAGE_VER}-cp35-cp35m-linux_x86_64.whl
+ export TENSORFLOW_BIN_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-${TENSORFLOW_PACKAGE_VER}-cp35-cp35m-linux_x86_64.whl
else
- export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-${TENSORFLOW_PACKAGE_VER}-cp35-cp35m-linux_x86_64.whl
+ export TENSORFLOW_BIN_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-${TENSORFLOW_PACKAGE_VER}-cp35-cp35m-linux_x86_64.whl
fi
- pip3 install --upgrade $TF_BINARY_URL -t ${TENSORFLOW_LIB_DIR}
+ pip3 install --upgrade ${TENSORFLOW_BIN_URL} -t ${TENSORFLOW_LIB_DIR} --trusted-host storage.googleapis.com
fi
######################################################################################
