File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 1- -f https://download.pytorch.org/whl/torch_stable.html
2- -f https://download.pytorch.org/whl/torch/
31--extra-index-url https://download.pytorch.org/whl/cu113
4- torch == 1.11.0+cu113
2+ torch == 1.11.0
53pybind11 == 2.6.2
Original file line number Diff line number Diff line change 11from torch_tensorrt import __version__
22from torch_tensorrt import _C
33
4+ import torch
5+
46
57def dump_build_info ():
68 """Prints build information about the torch_tensorrt distribution to stdout
@@ -15,7 +17,9 @@ def get_build_info() -> str:
1517 str: String containing the build information for torch_tensorrt distribution
1618 """
1719 build_info = _C .get_build_info ()
18- build_info = "Torch-TensorRT Version: " + str (__version__ ) + '\n ' + build_info
20+ build_info = "Torch-TensorRT Version: " + str (__version__ ) + '\n ' \
21+ + "Using PyTorch Version: " + str (torch .__version__ ) + '\n ' \
22+ + build_info
1923 return build_info
2024
2125
You can’t perform that action at this time.
0 commit comments