您好,欢迎来到化拓教育网。
搜索
您的当前位置:首页onnx模型预测环境安装笔记

onnx模型预测环境安装笔记

来源:化拓教育网

环境部署

1、onnxruntime安装

CPU版本:

直接pip安装

pip install onnxruntime

GPU版本:

先查看自己CUDA版本然后在下面的链接去找对应的onnxruntime的版本

查询到对应版本,直接pip安装即可,例如

pip install onnxruntime-gpu==1.13.1

2、环境验证

输入一下代码

import onnxruntime

print(onnxruntime.get_device())

ort_session = onnxruntime.InferenceSession("model_data/to46_cut.onnx",providers=['CUDAExecutionProvider'])

print(ort_session.get_providers())

结果显示是下面这样的就代表成功了

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- huatuo9.cn 版权所有 赣ICP备2023008801号-1

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务