No module named torch.fx (torch 1.7.0)

Hi @aysekonus, which version of JetPack-L4T are you running? You can check your L4T version with cat /etc/nv_tegra_release

Are you using the PyTorch wheels from this thread? https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

Have you tried testing the l4t-pytorch containers to see if you also have the issue there? https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-pytorch