set environment variable for Anaconda's numpy 1.17

Author: njzjz

source/train/env.py

@@ -18,9 +18,13 @@ 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__:
+    if 'mkl_info' in np.__config__.__dict__ or 'mkl_rt' in np.__config__.blas_mkl_info['libraries']:
         set_env_if_empty("KMP_BLOCKTIME", "0")
         set_env_if_empty("KMP_AFFINITY", "granularity=fine,verbose,compact,1,0")
         reload(np)