onnx模型预测环境安装笔记_pip安装onnxruntime

环境部署

1、onnxruntime安装

CPU版本：

直接pip安装

pip install onnxruntime

GPU版本：

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

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())

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