Deploy GNN model to tensorrt c++ platform


I had trained a graph neural networks model with pytorch_geometric module, and now I want to deploy the model to tensorrt C++ platform, but I don’t find some relevant conversion tips, could you so nice to give me some guidance?


TensorRT Version: 8.0.1
GPU Type: 3090
Nvidia Driver Version: 515
CUDA Version: 11.1
CUDNN Version: 8.2
Operating System + Version: ubuntu18.04
Python Version (if applicable): 3.8
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 1.8.0
Baremetal or Container (if container which image + tag):

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered


Please refer to the following developer guide and samples which may help you.

Thank you.