torchviz module not found

[egmaminta]

Good day! I just discovered this handy package. I followed the instructions (installed thru pip install torchviz, etc.) but when I run my code, it produces an error saying that torchviz module is not found. I was hoping you could enlighten me on this. Thank you!

[albanD]

Hi,

This usually happens because you installed with pip for a different version of python compared to the one where you try to run the code. You can avoid this by doing:

python -m pip install torchviz
python
>>> import torchviz # That one should work fine

[egmaminta]

I see. Thank you!