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


发布评论