No OpKernel was registered to support Op TRTEnginOp used by {{node_TRTEngineOp_4}}

ardianumam / Tensorflow-TensorRT

This repository is for my YT video series about optimizing a Tensorflow deep learning model using TensorRT. We demonstrate optimizing LeNet-like model and YOLOv3 model, and get 3.7x and 1.5x faster for the former and the latter, respectively, compared to the original models.

303 stars 110 forks source link

No OpKernel was registered to support Op TRTEnginOp used by {{node_TRTEngineOp_4}} #33

Open Niloofar-bgh opened 3 years ago

Niloofar-bgh commented 3 years ago

Running

2_inference_using_TensorRT-model.ipynb

gives the below error, and I have:

The TensorFlow-GPU 1.15.0
The Nvidia RTX 2070

Any idea what could be the reason?!

anantgupta129 commented 3 years ago

@Niloofar-bgh I am facing same error did you find any solution to this?

Niloofar-bgh commented 3 years ago

@Niloofar-bgh I am facing same error did you find any solution to this?

Not really; I was trying to run the code on my personal computer, I wasn't successful, then I switched to using Google Colab, which I successfully executed the code! My assumption for the issue is the version incompatibility of libraries and dependencies! However, if you found a solution to the problem, please let me know.