Closed udit7395 closed 3 months ago
@udit7395 I think the issue is that the Jetson's hardware JPEG decoder uses a different API (either through GStreamer or the Jetson Multimedia API), and doesn't use that same nvjpeg.h interface. I can't find nvjpeg.h on my Jetson's either...
That makes sense, thanks @dusty-nv. I don't think there is anything that can be done at the container level, so please feel free to close this issue.
For future reference, I raised an issue on the Jetson forums about this.
torchvision supports decoding jpeg image via nvjpeg but I get an error when using it on Jetson AGX Orin development kit
Code
Error
Container Used: dustynv/l4t-pytorch:r35.4.1
I suspect while building torchvision, it was unable to find nvjpeg. This might be because it is trying to look for nvjpeg in
CUDA_HOME
(/usr/local/cuda) which is true of x86 based systems but not for Jetson's I guess. I ran the find command to find "nvjpeg.h" but was unable to find its location on the Orin. Could you help with this issue? I would like build torchvision with nvJPEG as it helps loads image faster instead of loading image via CPU using opencv and then moving to GPUSnippet of code used to build torchvision to find nvjpeg