neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs
https://neuralmagic.com/deepsparse/
Other
3.01k stars 176 forks source link

Please install deepsparse[yolo] to use this pathway #1091

Closed kopyl closed 1 year ago

kopyl commented 1 year ago

I installed deepsparse like this: pip install deepsparse[server,yolo,onnxruntime]

then launched python. Here is my code:

from deepsparse import Pipeline
model_stub="model.onnx"
yolo_pipeline = Pipeline.create(
    task="yolo",
    model_path=model_stub,
)

And got this error. Why? How can I fix it?

kopyl commented 1 year ago

I was able to make it work by removing opencv-python and installing opencv-python-headless