Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

需要添加 CUDA / cudnn / MKL 等版本的信息查询接口 #388

Open
bealwang opened this issue Nov 10, 2021 · 0 comments
Open

需要添加 CUDA / cudnn / MKL 等版本的信息查询接口 #388

bealwang opened this issue Nov 10, 2021 · 0 comments

Comments

@bealwang
Copy link

bealwang commented Nov 10, 2021

背景

  • 由于 TensorRT 不同版本 dump 的模型并不兼容。假如一个用户想用 MGE 做 TensorRT 模型的推理,此时他需要知道 MGE 中的 TensorRT 版本是什么。
  • 当用户发现模型精度或速度对不齐时,无论是 MGE 多个版本之间对不齐还是和其他框架对不齐,用户可能需要知道 MGE 使用的 CUDA Toolkit 的版本,cuDNN,MKL 等的版本。
  • 当用户使用 MGE 并且想指定显卡的时候,需要知道当前在哪张卡上,环境中总共有几张卡,这些卡的信息例如 name,compute capability 等信息。

需求描述

实现包括但不限于以下功能

  • mge.version.cuda
  • mge.backends.cudnn.version()
  • mge.cuda.xx
@xxr3376 xxr3376 changed the title Feature Request 需要添加 CUDA / cudnn / MKL 等版本的信息查询接口 Apr 1, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants