search

onnx / tensorflow-onnx

Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Apache License 2.0
2.33k stars 432 forks source link

how to use onnx model to process torch.Tensor data on GPU #2363

Open yangqiheng2019 opened 2 weeks ago

yangqiheng2019 commented 2 weeks ago

Ask a Question

Question

Further information

Model opset:

Notes

yangqiheng2019 commented 2 weeks ago

Now,I use yolov7 onnx model to process ,but it must to.cpu() ,it means I can not use GPU to process data, it will spend more time.

fatcat-z commented 2 weeks ago

Looks like your issue is not related to converting a TF model to ONNX which is what this tool does.

yangqiheng2019 commented 2 weeks ago

This model was converted from PT