Hello, I want to process images with a GPU using the NVIDIA Xavier NX AI kit. I have installed torch. But where can I download torchvision and torchaudio in a way that is compatible with the GPU? Is there a direct download method using pip, instead of Docker?
You could build these packages as described here or you could use a container which ships with all pre-built libs.