From fe0eb0a5bbc8bf816d5649bfdd34908c258eb245 Mon Sep 17 00:00:00 2001
From: yancong <32220263+ice-tong@users.noreply.github.com>
Date: Tue, 30 Aug 2022 08:41:08 +0800
Subject: [PATCH] fix(CircleCI): Set MKL_THREADING_LAYER=GNU in dockefile
 (#484)

---
 .circleci/docker/Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/.circleci/docker/Dockerfile b/.circleci/docker/Dockerfile
index d9cf8cc7..7212c32e 100644
--- a/.circleci/docker/Dockerfile
+++ b/.circleci/docker/Dockerfile
@@ -4,6 +4,10 @@ ARG CUDNN="7"
 
 FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
 
+# Set MKL_THREADING_LAYER=GNU to fix issue:
+# https://github.com/pytorch/pytorch/issues/37377
+ENV MKL_THREADING_LAYER GNU
+
 # To fix GPG key error when running apt-get update
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
-- 
GitLab