Как вычислить градиент относительно состояния периодической сети на каждом временном шаге?

Я получил ошибку при использовании matplotlib через Spark. matplotlib.use('Agg') не работает для меня. В конце концов, для меня работает следующий код. Подробнее здесь

import matplotlib.pyplot as plt.
plt.switch_backend('agg')