Can jetson use pytorch geometric lib to train GNN models?

Device: Jetson AGX Xavier
JetPack: 4.6
This link (PyTorch for Jetson ) provided “pre-built PyTorch pip wheel” is using python3.6, but PyG is available for Python 3.7 to Python 3.11.
I want to train GNN models on Jetson AGX Xavier, what should I do?

Hi @1911047, since you are on Jetson Xavier, you could upgrade your version of JetPack to JetPack 5.x, which has PyTorch wheels available for Python 3.8. Alternatively, you could rebuild PyTorch for your desired version of Python - see the Build from Source section of this post:

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.