MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_tensorrt_version

Function get_tensorrt_version

imperative/python/megengine/device.py:325–331  ·  view source on GitHub ↗

r"""Get the TensorRT version used when compiling MegEngine. Returns: a version number, indicating `NV_TENSORRT_MAJOR * 1000 + NV_TENSORRT_MINOR * 100 + NV_TENSORRT_PATCH`.

()

Source from the content-addressed store, hash-verified

323
324
325def get_tensorrt_version():
326 r"""Get the TensorRT version used when compiling MegEngine.
327
328 Returns:
329 a version number, indicating `NV_TENSORRT_MAJOR * 1000 + NV_TENSORRT_MINOR * 100 + NV_TENSORRT_PATCH`.
330 """
331 return _get_tensorrt_version()
332
333
334def get_cuda_driver_version():

Callers 1

init_commonFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected