PyTorch can be installed with:
https://pytorch.org/get-started/locally/
Ideally, you would run PyTorch from Jupyter:
Also, the “graphviz” python package is handy to investigate what is going on inside PyTorch:
https://graphviz.readthedocs.io/en/stable/manual.html#installation