diff --git a/requirements.txt b/requirements.txt
index fea6b09..d590f33 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ torch==1.0.1.post2
 torchvision==0.2.2
 prettytable==0.7.2
 numpy==1.16.2
-tqdm==4.29.1
+tqdm==4.19.9
 coloredlogs==10.0
 scipy==1.2.1
 setproctitle==1.1.10
@@ -15,3 +15,4 @@ tensorflow_gpu==1.9.0
 matplotlib==2.0.2
 visdom==0.1.8.8
 Cython==0.29.6
+scikit-learn==0.22.1
diff --git a/src/m_utils/visualize.py b/src/m_utils/visualize.py
index 3bd89a4..f6f4e2d 100644
--- a/src/m_utils/visualize.py
+++ b/src/m_utils/visualize.py
@@ -1,4 +1,8 @@
-
+import os
+import matplotlib as mpl
+if os.environ.get('DISPLAY','') == '':
+    print('no display found. Using non-interactive Agg backend')
+    mpl.use('Agg')
 import matplotlib.pyplot as plt
 import seaborn as sns
 from PIL import Image