Unable to install Torch compatible with Jetson 5.1 and CUDA 11.4

Robotics & Edge Computing Jetson & Embedded Systems Jetson Xavier NX
1

Hello,
We are currently working with a Jetson Xavier NX to train a vision model with a rather large dataset. We are trying to get GPU access from within our docker container to speed up this process, but CUDA 11.4 doesn’t work without the compatible version of Torch that can use this version of CUDA. We’ve already tried installing torch using developer docs. Any suggested fixes?

3

Hi,

Are you using a custom Python version?

The package shared in our doc should have CUDA enabled:

But if you have a custom Python version, ex 3.9, please build PyTorch from the source with GPU enabled.

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.