I need to install torch 2.0.1 for NVidia AGX Orin 64GB dev kit but not able to find it. Can anyone share the link please. If I am install using pip install torch then torch.cuda.is_available() return false but when installing from here https://developer.download.nvidia.com/compute/redist/jp/v51/pytorch/torch-2.0.0a0+8aa34602.nv23.03-cp38-cp38-linux_aarch64.whl I can get the cuda device but then I am not able to install torchvision and torchaudio packages. Hope to get response soon. thanks
Your topic was posted in the wrong category. I am moving this to the Jetson AGX Orin forum for visibility.
Do you meet errors when installing torchvision and torchaudio ?
Could you share the error log with us?
Or you can try our
It has PyTorch, torchvision and torchaudio installed.
BTW torch 2.0.1 wheel is found here if you require it:
I meant there isn’t any error except that I am not able to find the compatible version of torchvision and torchaudio. see below
ERROR: torchvision 0.15.2 has requirement torch==2.0.1, but you’ll have torch 2.0.0a0+8aa34602.nv23.3 which is incompatible.
ERROR: torchaudio 2.0.2 has requirement torch==2.0.1, but you’ll have torch 2.0.0a0+8aa34602.nv23.3 which is incompatible.
@iqbal.rahmoon I believe these message imply that you are trying to install torchvision/torchaudio via pip and their wheels on PyPi, which won’t work anyways because those PyPi wheels are built without CUDA support. Instead, see the
Installation section of this post to build torchvision from source, or see the l4t-pytorch dockerfile to see how I build them:
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.