torchnvjpeg
torchnvjpeg copied to clipboard
Decode JPEG image on GPU using PyTorch
When this code is using a multi-card, it directly uses no CUDA Context environment switch, and pytorch uses the last environment by default, so the decoder created in front of...
Signle decode cost 137 ms per image Batch decode cost 0.64 ms per image I guess the reason is the torchnvjpeg.Decoder parameter different, so I set the param without torch.cuda.current_stream(device_id),...
Hello!.I got error when install the package!Can you help me? torchnvjpeg.Decoder()
I can run single decode and batch decode perfectly fine but parallel decode give me this error ``` { "name": "RuntimeError", "message": "Caught an unknown exception!", "stack": "--------------------------------------------------------------------------- RuntimeError Traceback...