f2889ca90ddd60b19070b7e8fbcb0867ae4c47ed,setup.py,,,#,25

Before Change

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"

After Change

gpu_available = len(out) > 0

// 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
if "linux" in sys.platform:
    system_type = "linux"

Instances