diff --git a/distiller/data_loggers/collector.py b/distiller/data_loggers/collector.py
index fc96aa0800072ae1c64d224b50d4bb77cf5629b6..c859924ff9345287fc0842a7f357acc5da054d98 100755
--- a/distiller/data_loggers/collector.py
+++ b/distiller/data_loggers/collector.py
@@ -26,6 +26,8 @@ import torch
 from torchnet.meter import AverageValueMeter
 import logging
 from math import sqrt
+import matplotlib
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import distiller
 msglogger = logging.getLogger()