File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed
Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff 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.
Original file line number Diff line number Diff 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.
You can’t perform that action at this time.
0 commit comments