We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 81d6bcc commit c3a65efCopy full SHA for c3a65ef
py/torch_tensorrt/__init__.py
@@ -4,10 +4,11 @@
4
import sys
5
from typing import Dict, List
6
7
-from torch_tensorrt._version import (
+from torch_tensorrt._version import ( # noqa: F401
8
__cuda_version__,
9
__cudnn_version__,
10
__tensorrt_version__,
11
+ __version__,
12
)
13
14
from packaging import version
@@ -87,6 +88,7 @@ def _find_lib(name: str, paths: List[str]) -> str:
87
88
from torch_tensorrt._enums import * # noqa: F403
89
from torch_tensorrt._Input import Input # noqa: F401
90
from torch_tensorrt._utils import * # noqa: F403
91
+from torch_tensorrt._utils import sanitized_torch_version
92
93
if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
94
from torch_tensorrt import dynamo # noqa: F401
0 commit comments