Hi,
Please noted that you would need to install the package built with the same JetPack version.
We don’t have a PyTorch 1.4.0 for JetPack4.4, but only for JetPack 4.4 DP.
Do you have any dependency on PyTorch 1.4.0?
If not, you can try the v1.6.0 and v1.7.0 shared on the same website.
JetPack 4.4 / 4.4.1
- PyTorch v1.6.0 pip wheel (Python 3.6)
- PyTorch v1.7.0 pip wheel (Python 3.6)
If yes, you can build it from source with the instructions shared below:
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-7-0-now-available/72048
Another alternative is to use l4t-pytorch docker that has PyTorch pre-installed.
Thanks.