Visualisation de réseaux sur Jupyter notebook avec plot_model

IFB Core Cluster
,
1

Bonjour,

Je cherche à visualiser des réseaux de neurones avec plot_model de keras.utils. J'y arrive hors de jupyter notebook, en faisant par exemple dans mobaxterm :

module load python-pytorch-tensorflow/3.9-1.11.0-2.6.2
module load graphviz/2.40.1
python

puis

import numpy as np
from keras.models import Sequential
from keras.layers import Dense
from keras.utils import plot_model

# Create a simple MLP model
model = Sequential([
    Dense(64, activation='relu', input_shape=(100,)),
    Dense(32, activation='relu'),
    Dense(10, activation='softmax')
])

# Compile the model
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

# Visualize the model architecture
plot_model(model, to_file='model_plot.png', show_shapes=True, show_layer_names=True)

Par contre la même chose dans jupyter notebook donne l'erreur :
"You must install pydot (pip install pydot) and install graphviz"

J'ai graphviz, pydot et keras.utils dans /shared/ifbstor1/home/dcros/.local/lib/python3.9/site-packages, et j'utilise un notebook avec :

print(sys.executable)
/shared/ifbstor1/software/miniconda/envs/python-pytorch-tensorflow-3.9-1.11.0-2.6.2/bin/python
!{sys.executable} --version
Python 3.9.15

Merci

David