fix: Add version checking for torch._dynamo import in __init__ (#1881)

Author: gs-olive

py/torch_tensorrt/__init__.py

@@ -4,6 +4,7 @@
 import sys
 import platform
 import warnings
+from packaging import version
 from torch_tensorrt._version import (
     __version__,
     __cuda_version__,
@@ -93,8 +94,10 @@ def _find_lib(name, paths):
 from torch_tensorrt._TRTModuleNext import TRTModuleNext
 
 from torch_tensorrt import fx
-from torch_tensorrt import dynamo
-from torch_tensorrt.dynamo import torch_compile
+
+if version.parse(torch.__version__) >= version.parse("2.dev"):
+    from torch_tensorrt import dynamo
+    from torch_tensorrt.dynamo import torch_compile
 
 
 def _register_with_torch():