diff --git a/wgan.py b/wgan.py
index 95cf6db94cd7be2a528a7f695560cc926bdd1775..19360a9129455e20de7031fae63b835b8116a3e2 100644
--- a/wgan.py
+++ b/wgan.py
@@ -12,6 +12,8 @@ from torch.utils.data import DataLoader
 from torch.autograd import Variable
 
 
+os.environ['MKL_THREADING_LAYER'] = 'GNU'
+
 torch.autograd.set_detect_anomaly(True)
 
 sys.path.append('/opt/regressor/src')