The reason is that the .whl files in https://download.pytorch.org/whl/cu121 are for Linux (x86_64).
Solution: Follow the instructions from NVIDIA Developer Forums - PyTorch for Jetson to install the correct .whl files (linux_aarch64).
The issue is resolved.
import torch
printf(torch.version)
Traceback (most recent call last):
File “”, line 1, in
NameError: name ‘printf’ is not defined. Did you mean: ‘print’?
print(torch.version)
2.3.0
print(torch.cuda.is_available())
True
print(torch.version.cuda)
12.2