However when i was running my code i realised that it wasnt installed with CUDA. So if I look for the torch version, I see version 1.12.0 installed.
However, if i try to run this command, print(torch.cuda.is_available()) it returns false. Have I installed it wrongly ? If so, can anyone point me to the right way of installing pytorch on the nano?
Hi @archieprad, I don’t believe that we have a PyTorch+CUDA arm64 wheel built for Nano/JetPack 4.x and PyTorch 1.12, so my guess is the one that you installed was built without CUDA enabled.
Hi @dusty_nv thanks for the suggestion! I tried it but I got this error ERROR: torch-1.10.0-cp36-cp36m-linux_aarch64.whil is not a supported wheel on this platform. I actually get this error for all the versions that i tried from the link
We only provide pre-built PyTorch pip wheels for the default version of Python that comes with Ubuntu - on JetPack 4.x, that’s Ubuntu 18.04 and Python 3.6. However, you can find the instructions for building PyTorch from source in that PyTorch topic.