in first, i install torch 1.8.0 but when i have “RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED - PyTorch” error, i install torch 1.7.0 then it solved. I read something like that: “torch.fx comes with torch 1.8.0” but i cant upgrade because of “RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED - PyTorch” error.
You may want to try upgrading to a more recent version of JetPack or re-flashing your device to see if that fixes the issue.
The containers come with the components pre-installed and can help discern if it’s a PyTorch install issue on your side, or if it is something with your device setup (i.e. cuDNN). You can pick one of the container tags that matches your L4T version and follow the instructions on this page to run it: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-pytorch