环境部署
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())
结果显示是下面这样的就代表成功了