Source code for torch_tensorrt._utils
from typing import Any
import torch
from torch_tensorrt import _C
from torch_tensorrt._version import __version__
[docs]def dump_build_info() -> None:
"""Prints build information about the torch_tensorrt distribution to stdout"""
print(get_build_info())
[docs]def get_build_info() -> str:
"""Returns a string containing the build information of torch_tensorrt distribution
Returns:
str: String containing the build information for torch_tensorrt distribution
"""
core_build_info = _C.get_build_info()
build_info = str(
"Torch-TensorRT Version: "
+ str(__version__)
+ "\n"
+ "Using PyTorch Version: "
+ str(torch.__version__)
+ "\n"
+ core_build_info
)
return build_info
def sanitized_torch_version() -> Any:
return (
torch.__version__
if ".nv" not in torch.__version__
else torch.__version__.split(".nv")[0]
)