2. import torch 
    3. from torch import onnx
    6. dummy_input = torch.randn(10,3,256,256).cuda()
    7. path = './net.pt'
    8. model = torch.load(path)
    9. input_names = ['input']
    10. output_names = ['output']
    11. torch.onnx.export(model, dummy_input,'net.onnx' ,verbose=True, input_names=input_names, output_names=output_names,opset_version=11)

    opset_version=11
    版本要低 ，最好为9，这样onnxruntime 版本低 接口兼容。

    onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: input for the following indices