search

Unstructured-IO / unstructured-api

Apache License 2.0
429 stars 94 forks source link

Why do I get "Cannot Initialize NVML"? How can I initialize the GPU kernel? #418

Open Bryson14 opened 1 month ago

Bryson14 commented 1 month ago

Describe the bug A clear and concise description of what the bug is.

When I run unstrctured docker container locally, I get this error:

2024-05-20 12:44:22 /home/notebook-user/.local/lib/python3.10/site-packages/torch/cuda/init.py:619: UserWarning: Can't initialize NVML 2024-05-20 12:44:22 warnings.warn("Can't initialize NVML")

I assume that its because its running the CPU version of unstructured and not the GPU version. How can i fix this? running in a GPU would help with hi-res parsing

To Reproduce Please provide as much info as possible:

Environment:

Additional context Add any other context about the problem here.

MthwRobinson commented 1 month ago

@Bryson14 - Could you provide more context on the environment you're running this in?