By following this guide, I installed Pytorch 1.7.0 on a Nvidia Jetson Nano, which had JetPack 4.4 and Python 3.6 installed. But it seems like the installation was not proper.
Normally Pytorch outputs the following:
>> torch.tanh(torch.tensor(4.2)) >> torch.tensor(1.)
But in the current Jetson Nano installation:
>> torch.tanh(torch.tensor(4.2)) >> torch.tensor(4.)
In other test scripts, the models do not seem to update their weights,
which usually works in Pytorch.
How should I install Pytorch on the Jetson Nano? I already tried the corresponding wheel.