Skip to content
This repository was archived by the owner on Nov 8, 2022. It is now read-only.

Commit cbf4ad8

Browse files
authored
Peter/mkl install fix (#341)
* fix tf mkl install
1 parent e2dc01b commit cbf4ad8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,8 @@
9696
tf_be = os.getenv('NLP_ARCHITECT_BE', False)
9797
if tf_be and 'mkl' == tf_be.lower():
9898
if py3_ver == 5 or py3_ver == 6:
99-
chosen_tf = tf_mkl_url.format(tf_version, py3_ver, py3_ver)
99+
tf_mkl_url_real = tf_mkl_url.format(tf_version, py3_ver, py3_ver)
100+
subprocess.run('pip3 install -U {}'.format(tf_mkl_url_real), shell=True)
100101
elif tf_be and 'gpu' == tf_be.lower() and gpu_available:
101102
chosen_tf = 'tensorflow-gpu=={}'.format(tf_version)
102103
requirements.append(chosen_tf)

0 commit comments

Comments
 (0)