Hi @bangwhe, PyTorch 1.7 had a bug on aarch64 which led to incorrect tensor results on the CPU (see PyTorch issue #47098)
This issue was patched with this patch, which is linked to in the Build Instructions from the PyTorch topic: https://gist.github.com/dusty-nv/ce51796085178e1f38e3c6a1663a93a1#file-pytorch-1-7-jetpack-4-4-1-patch
So that wheel you are using was probably built with this patch applied to fix the issue.