Skip to content

Commit 1d58a7c

Browse files
committed
Changing the tensorrt.bindings.plugin imports to tensorrt.plugin and changing training/vgg16/requirements.txt
1 parent 72c9163 commit 1d58a7c

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed
Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
tensorboard>=1.14.0
22
protobuf==3.20.*
33
nvidia-pyindex
4-
--extra-index-url https://pypi.nvidia.com
54
tqdm
65
nvidia-modelopt
7-
--extra-index-url https://pypi.nvidia.com
6+

py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
import numpy as np
55

66
# Seems like a bug in TensorRT
7-
import tensorrt_bindings.plugin as trtp
7+
import tensorrt.plugin as trtp
88
import torch
9-
from tensorrt_bindings.plugin._lib import QDP_REGISTRY
9+
from tensorrt.plugin._lib import QDP_REGISTRY
1010
from torch.fx.node import Argument, Node, Target
1111
from torch_tensorrt.dynamo._settings import CompilationSettings
1212
from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext

0 commit comments

Comments
 (0)