Skip to content

Commit

Permalink
docs(tensorrt/utils.py): update docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
tpoisonooo committed Sep 5, 2022
1 parent 2c1b432 commit 76606e8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmdeploy/backend/tensorrt/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def load(path: str) -> trt.ICudaEngine:


def search_cuda_version() -> str:
"""try import torch.version.cuda, then try use cmd.
"""try cmd to get cuda version, then try `torch.cuda`

Returns:
str: cuda version, for example 10.2
Expand Down

0 comments on commit 76606e8

Please sign in to comment.