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

[yangqiheng2019]

Ask a Question

Question

Further information

• Is this issue related to a specific model?
  Model name:

Model opset:

Notes

[yangqiheng2019]

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]

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

[yangqiheng2019]

This model was converted from PT

[Inquisitive-ME]

This library is for converting tensorflow models to onnx you are in the wrong repo for your question