if tf_be and "mkl" == tf_be.lower():
if py3_ver == 5 or py3_ver == 6:
tf_mkl_url_real = tf_mkl_url.format(tf_version, py3_ver, py3_ver)
subprocess.call([sys.executable, "-m", "pip", "install", tf_mkl_url_real])
elif tf_be and "gpu" == tf_be.lower() and gpu_available:
chosen_tf = "tensorflow-gpu=={}".format(tf_version)
for r in install_requirements:
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
