Skip to content

Commit 7dfad50

Browse files
committed
add docstring
Signed-off-by: YunLiu <[email protected]>
1 parent b55bf6a commit 7dfad50

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

monai/bundle/scripts.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1589,6 +1589,8 @@ def trt_export(
15891589
"""
15901590
Export the model checkpoint to the given filepath as a TensorRT engine-based TorchScript.
15911591
Currently, this API only supports converting models whose inputs are all tensors.
1592+
Note: NVIDIA Volta support (GPUs with compute capability 7.0) has been removed starting with TensorRT 10.5.
1593+
Review the TensorRT Support Matrix for which GPUs are supported.
15921594
15931595
There are two ways to export a model:
15941596
1, Torch-TensorRT way: PyTorch module ---> TorchScript module ---> TensorRT engine-based TorchScript.

monai/networks/trt_compiler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -505,7 +505,9 @@ def trt_compile(
505505
) -> torch.nn.Module:
506506
"""
507507
Instruments model or submodule(s) with TrtCompiler and replaces its forward() with TRT hook.
508-
Note: TRT 10.3 is recommended for best performance. Some nets may even fail to work with TRT 8.x
508+
Note: TRT 10.3 is recommended for best performance. Some nets may even fail to work with TRT 8.x.
509+
NVIDIA Volta support (GPUs with compute capability 7.0) has been removed starting with TensorRT 10.5.
510+
Review the TensorRT Support Matrix for which GPUs are supported.
509511
Args:
510512
model: module to patch with TrtCompiler object.
511513
base_path: TRT plan(s) saved to f"{base_path}[.{submodule}].plan" path.

0 commit comments

Comments
 (0)