Hello @jotapedromabreu and welcome to the NVIDIA developer forums!
I highly recommend going through the “Getting Started” instructions on Start Locally | PyTorch and check that you are fulfilling all prerequisites.
In addition to that I will move your post to the Jetson Orin category, they will haave better insight into what requirements are needed to run pytorch with CUDA on that device.