https://github.com/eastonman/tensorrt-pytorch-wrapper

A wrapper makes TensorRT engine accept PyTorch Cuda Tensor.
https://github.com/eastonman/tensorrt-pytorch-wrapper

cuda pytorch tensorrt

Last synced: about 1 month ago
JSON representation

A wrapper makes TensorRT engine accept PyTorch Cuda Tensor.

• Host: GitHub
• URL: https://github.com/eastonman/tensorrt-pytorch-wrapper
• Owner: eastonman
• License: agpl-3.0
• Created: 2021-10-15T13:25:49.000Z (about 3 years ago)
• Default Branch: main
• Last Pushed: 2021-10-15T14:13:36.000Z (about 3 years ago)
• Last Synced: 2023-03-04T18:07:35.701Z (almost 2 years ago)
• Topics: cuda, pytorch, tensorrt
• Language: Python
• Homepage:
• Size: 18.6 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0

Awesome Lists containing this project