我正在尝试运行一个ONNX模型。
import onnxruntime as ort
import onnxruntime.backend
model_path = "model.onnx"
#https://microsoft.github.io/onnxruntime/
ort_sess = ort.InferenceSession(model_path)
print( ort.get_device() )
这将打印出来
cpu
如何让它在我的GPU上运行?如何确认它正在工作?
python -c "import onnxruntime as ort;print(ort.get_device())"
- undefined