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.

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

Cannot find the engine #964

Open
PredyDaddy opened this issue Apr 25, 2024 · 0 comments
Open

Cannot find the engine #964

PredyDaddy opened this issue Apr 25, 2024 · 0 comments

Comments

@PredyDaddy
Copy link

I used this sample to infer an onnx from yolox,

import onnx
import onnx_tensorrt.backend as backend
import numpy as np

model = onnx.load("/path/to/model.onnx")
engine = backend.prepare(model, device='CUDA:1')
input_data = np.random.random(size=(1, 3, 640, 640)).astype(np.float32)
output_data = engine.run(input_data)[0]
print(output_data)
print(output_data.shape)

I can see the from command windows that tensorrt is generating the enigne, and later it shows the output shape is this same as my onnx, but I cannot find the engine where it generate, which means every time I run the code, I need generating the engine again.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant