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"
tf_mkl_url = "https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-{}-cp3{}-cp3{}m-linux_x86_64.whl"

After Change

    system_type = "linux"
    tf_be = os.getenv("NLP_ARCHITECT_BE", False)
    if tf_be and "mkl" == tf_be.lower():
        chosen_tf = "intel-tensorflow=={}".format(tf_version)
    elif tf_be and "gpu" == tf_be.lower() and gpu_available:
        chosen_tf = "tensorflow-gpu=={}".format(tf_version)

for r in install_requirements:

Instances