Hi,
You can find the package for Jetson on the following topic:
Or our NGC container: l4t-pytorch or l4t-ml.
Thanks.