2023年11月29日发(作者:)

pytorch转换onnxruntime

特点:

onnxruntime⽀持if,动态输⼊

环境准备:

1.

pip install onnx==1.9.0pip install onnxruntime-gpu==1.9.0

2. 报错 [ONNXRuntimeError] : 9 : NOT_IMPLEMENTED : Could not find an implementation for xxx node with name 'yyy'

解决:

onnx 版本对应关系 ,在本机安装对应版本的 CUDA cudnn

3. 报错 .1.9.0: cannot open shared object file: No such file or dictory

解决:

import torch