TensorRT
TensorRT copied to clipboard
How to accelerate GCN using torch-geometric with TensorRT?
You can export onnx, then build/infer by trtexec.
Also take a look at https://github.com/pytorch/TensorRT. Closing for now since no response yet but free to reopen the issue!