How to use the GPU version?

Unstructured-IO / unstructured-api

Apache License 2.0

528 stars 110 forks source link

How to use the GPU version? #289

Closed homeant closed 11 months ago

homeant commented 1 year ago

How to use ONNX with CUDA for recognition?

awalker4 commented 11 months ago

Hi there, to use CUDA you'll need to switch out ONNX for the onnxruntime-gpu package. If you're running on a gpu instance with all the right nvidia drivers and utils, I believe this should work.