From aae4eb098c1870a1ecb4bd081d42e5ebe88368b0 Mon Sep 17 00:00:00 2001
From: Philippe Modard <philmod@google.com>
Date: Tue, 8 Feb 2022 10:29:52 -0500
Subject: [PATCH] Set KMP_SETTINGS to false to avoid Tensorflow warning message

http://b/218492749
---
 Dockerfile.tmpl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index d0de7ad1..4d2458a3 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -39,7 +39,7 @@ ARG TORCHVISION_VERSION
 ENV KMP_WARNINGS=0
 # Also make the KMP logs noverbose.
 # https://stackoverflow.com/questions/70250304/stop-tensorflow-from-printing-warning-message
-ENV KMP_AFFINITY="granularity=fine,noverbose,compact,1,0"
+ENV KMP_SETTINGS=false
 
 ADD clean-layer.sh  /tmp/clean-layer.sh
 ADD patches/nbconvert-extensions.tpl /opt/kaggle/nbconvert-extensions.tpl