A clear and concise description of the bug or issue.
TensorRT Version: 184.108.40.206
GPU Type: Quadro RTX 4000
Nvidia Driver Version: 582.02
CUDA Version: 11.6
CUDNN Version: 8.4
Operating System + Version: Windows 10
Python Version (if applicable): 3.6
TensorFlow Version (if applicable):
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):
Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)
- Exact steps/commands to build your repro
- Exact steps/commands to run your repro
- Full traceback of errors encountered
I am trying to port my CNN (densenet201), trained on tensorflow and converted in the ONNX format, on a new GPU after successfully using it on an old GPU (from the Quadro familly too).
However after converting the ONNX model to a .trt file with the following command : trtexec --onnx=model.onnx --saveEngine=densenet201.trt ; and loading into a code I developped following the Samples provided by NVIDIA, i always get the same output (9 output tensor) → [0,0,1,0,0,0,0,0,0].
Is there a specific version of TensorRT I have to download to make it work ?
You can find the code I use in the linked files.
LibTensorRT.cpp (10.5 KB)