Pytorch pth to onnx
[TOC]
ONNX环境配置Install1234# CPU版本pip install onnxruntime# GPU版本pip install onnxruntime-gpu
OnnxRun12345>>> import onnxruntime>>> onnxruntime.get_device()'GPU' #表示GPU可用>>> onnxruntime.get_available_providers()['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']
如果GPU不可用,可以在 ~/.bashrc 中添加下面两行内容:
12export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
Demo:
1234567891011121314151617import onnxruntim
...