tensorflow / tensorrt

TensorFlow/TensorRT integration
Apache License 2.0
736 stars 226 forks source link

about TF-TRT C++ Image Recognition Demo #337

Open yucathy opened 7 months ago

yucathy commented 7 months ago

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