Merge pull request #107 from njzjz/patch-2

set environment variable for Anaconda's numpy 1.17

Author: amcadmus

source/train/env.py

@@ -18,9 +18,15 @@ def set_env_if_empty(key, value):
 def set_mkl():
     """Tuning MKL for the best performance
     https://www.tensorflow.org/guide/performance/overview
+    
+    Fixing an issue in numpy built by MKL. See
+    https://github.com/ContinuumIO/anaconda-issues/issues/11367
+    https://github.com/numpy/numpy/issues/12374
     """
 
-    if 'mkl_info' in np.__config__.__dict__:
+    # check whether the numpy is built by mkl, see
+    # https://github.com/numpy/numpy/issues/14751
+    if 'mkl_rt' in np.__config__.get_info("blas_mkl_info").get('libraries', []):
         set_env_if_empty("KMP_BLOCKTIME", "0")
         set_env_if_empty("KMP_AFFINITY", "granularity=fine,verbose,compact,1,0")
         reload(np)