Skip to content

Bug Report: Hard-coded CUDA 12 check blocks CUDA 13 support in Torch-TensorRT #4438

Description

@joe0731

Bug Report: Hard-coded CUDA 12 check blocks CUDA 13 support in Torch-TensorRT

Bug Description

TensorRT-LLM has officially supported CUDA 13, but torch_tensorrt/_utils.py still contains a hard-coded version check that strictly restricts CUDA major version to 12.

The current logic blocks all CUDA 13 environments with the following error:

major, minor = map(int, cuda_version.split("."))
if major != 12:
    logger.error(
        "CUDA 13 is not currently supported for TRT-LLM plugins. Please install pytorch with CUDA 12.x support"
    )
    return False

This check is outdated and inconsistent with the latest TensorRT-LLM compatibility, preventing normal usage on CUDA 13 stacks.

To Reproduce

Steps to reproduce the behavior:

  1. Set up a full CUDA 13 environment
  2. Install PyTorch, TensorRT and TensorRT-LLM built for CUDA 13
  3. Initialize or run any workflow using Torch-TensorRT + TRT-LLM plugins
  4. The hard-coded version check fails and throws the CUDA 13 unsupported error

Expected Behavior

The version validation logic should be updated to recognize and allow CUDA 13. The error message should also be revised to match the latest supported CUDA versions. No functional restriction should exist if TensorRT-LLM natively supports CUDA 13.

Environment

  • Torch-TensorRT Version: N/A
  • PyTorch Version: N/A
  • CPU Architecture: x86_64
  • OS: Linux
  • PyTorch Installation Method: pip / conda / source
  • Build command: N/A
  • Source type: prebuilt / local source
  • Python version: N/A
  • CUDA version: 13.x
  • GPU models: N/A

Additional Context

This is purely a hard-coded version check issue rather than a real compatibility problem. TensorRT-LLM has added official CUDA 13 support, so the Torch-TensorRT validation layer needs to be synced and updated.

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions