I’m running Jetpack 6.0 on my Jetson Orin Nano. Ubuntu 22.04.4LTS. I used the cross platform installer from an Ubuntu host pc. This installed Cuda 12.2. I want to install torch and torchaudio using pip that is compatible with my device, Jetpack version, Cuda version and has Cuda support. I simply can’t find it.
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Aug_15_22:08:11_PDT_2023
Cuda compilation tools, release 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0
When I want to install pytorch with Cuda support, I can not get a torchaudio that works with the same torch version. I always get errors like such:
OSError: /home/tars/.local/lib/python3.10/site-packages/torchaudio/lib/libtorchaudio.so: undefined symbol: _ZN3c104cuda14ExchangeDeviceE
When I find a version that works together (2.3.0), it doesn’t have CUDA support.
import torch
torch.cuda.is_available()
returns False
I installed torch using the following command, which does offer CUDA support, but there’s no suitable torchaudio package. I always get the error above
wget https://developer.download.nvidia.com/compute/redist/jp/v60/pytorch/torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl -O torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl
pip3 install torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl
Is there a way to find the necessary files for this? I’d love to have some support