gpu_available = len(out) > 0
// check python version
py3_ver = int(platform.python_version().split(".")[1])
// Tensorflow version (make sure CPU/MKL/GPU versions exist before changing)
tf_version = "1.12.0"
tf_mkl_url = "https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-{}-cp3{}-cp3{}m-linux_x86_64.whl"
After Change
// Tensorflow version (make sure CPU/MKL/GPU versions exist before changing)
for r in install_requirements:
if r.startswith("tensorflow=="):
tf_version = r.split("==")[1]
// default TF is CPU
chosen_tf = "tensorflow=={}".format(tf_version)
// check system is linux for MKL/GPU backends
