IT博客汇
首页
精华
技术
设计
资讯
扯淡
权利声明
登录
注册
Python onnxruntime-gpu 指定 gpu 加载模型
Yiwei Zhang
发表于
2025-01-02 09:29:55
love
0
在 onnxruntime python 库中可以加载在内存或者显存中,分别使用 cpu 和 gpu 进行推断运算,本文记录 onnxruntime 将模型加载到指定 gpu 的步骤。