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

[Bryson14]

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:

• Filetype:
• Any additional API parameters:

Environment:

• Using the hosted API or self hosting?
• How are you calling the API? (Langchain, SDKs, cUrl, etc.)

Additional context
Add any other context about the problem here.

[MthwRobinson]

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