Как сохранить график на диск в python

Question

У меня проблемы с сохранением графиков (графиков) на диск в python. Я следую этому примеру, который находится на веб-сайте https://plot.ly/python/static-image-export/, но каждый раз, когда я запускаю код, я получаю эту ошибку

ValueError: Для создания изображения требуется пакет psutil .

Установка с использованием pip: $ pip install psutil

У меня установлен psutil на моей машине, его показ при запуске pip freeze

(dev ) рабочая станция: ~ $ pip freeze | grep psu

psutil == 5.7.0

Вот простой график, который я пытаюсь сохранить

import plotly.graph_objects as go
import numpy as np
np.random.seed(1)

N = 100
x = np.random.rand(N)
y = np.random.rand(N)
colors = np.random.rand(N)
sz = np.random.rand(N) * 30

fig = go.Figure()
fig.add_trace(go.Scatter(
    x=x,
    y=y,
    mode="markers",
    marker=go.scatter.Marker(
        size=sz,
        color=colors,
        opacity=0.6,
        colorscale="Viridis"
    )
))

fig.write_image("fig1.jpeg")

Буду признателен за помощь. Спасибо!

Платформа

Ubuntu 16.04

python3 .7