IT博客汇
  • 首页
  • 精华
  • 技术
  • 设计
  • 资讯
  • 扯淡
  • 权利声明
  • 登录 注册

    Python onnxruntime-gpu 指定 gpu 加载模型

    Yiwei Zhang发表于 2025-01-02 09:29:55
    love 0

    在 onnxruntime python 库中可以加载在内存或者显存中,分别使用 cpu 和 gpu 进行推断运算,本文记录 onnxruntime 将模型加载到指定 gpu 的步骤。



沪ICP备19023445号-2号
友情链接