tensorflow/tensorrt

about TF-TRT C++ Image Recognition Demo

Opened this issue · 0 comments

Thanks for your work. It's really helpful for me.

I just want to know whether it's possible to run the demo (TF-TRT C++ Image Recognition Demo) on a Jetson Orin NX by downloading the same image nvcr.io/nvidia/tensorflow:22.01-tf2-py3.
Because I would get some error output when I ran the python code, tf-trt-conversion.ipynb
when I want to run model = ResNet50(weights='imagenet')
the error is shown below
Non-OK-status: GpuLaunchKernel(FillPhiloxRandomKernelLaunch, num_blocks, block_size, 0, d.stream(), key, counter, gen, data, size, dist) status: INTERNAL: the provided PTX was compiled with an unsupported toolchain.
I'm not sure how to fix it.

device info
Capture