Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
yangqiheng2019 opened this issue Oct 29, 2024 · 3 comments
Open

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

yangqiheng2019 opened this issue Oct 29, 2024 · 3 comments
Labels
question An issue, pull request, or discussion needs more information

Comments

@yangqiheng2019
Copy link

Ask a Question

Question

Further information

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

Model opset:

Notes
@yangqiheng2019 yangqiheng2019 added the question An issue, pull request, or discussion needs more information label Oct 29, 2024
@yangqiheng2019
Copy link
Author

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
Copy link
Collaborator

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

@yangqiheng2019
Copy link
Author

This model was converted from PT

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
question An issue, pull request, or discussion needs more information
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@fatcat-z @yangqiheng2019