From 8f19741a829ca774d95d82faa7c00fd3877e4964 Mon Sep 17 00:00:00 2001
From: Paulo Scatena <phscatena@gmail.com>
Date: Wed, 18 Aug 2021 13:46:06 +0200
Subject: [PATCH] Fixing cuBLAS issue under Windows

---
 .../SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py b/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
index cbaf222ec..1b741cedd 100644
--- a/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
+++ b/AI4Animation/SIGGRAPH_2020/DeepLearning/Models/ExpertModel/main.py
@@ -41,7 +41,9 @@
 
 def main():
     rng  = np.random.RandomState(23456)
-    sess = tf.Session()
+    config = tf.ConfigProto()
+    config.gpu_options.allow_growth = True
+    sess = tf.Session(config=config)
     network = MainNN(
                 rng,
                 sess,