Hi @robinmorais, the normal torch wheels for aarch64 from PyTorch.org or pip/PyPi aren’t built with CUDA enabled, which is why they won’t detect the GPU. Those other PyTorch wheels that we provide are built with CUDA enabled.
Can you try uninstalling torchvision and building it from source like shown under the Installation section of this post: https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
For PyTorch 2.0, I’ve used torchvision v0.15.1, but for PyTorch 1.14 you might want to use torchvision v0.14.1 if you are having problems. Also if you continue having issues, you could try using the l4t-pytorch container which already has PyTorch/torchvision pre-installed.