peterwilli/sd-leap-booster

Only CPU usage on Windows

Closed this issue · 1 comments

Hey there, congratulations on these awesome results!
I'm trying to train an embedding on Windows, and while the installation was straightforward I can't seem to enable GPU usage. Maybe I'm missing something through the installation?

On further inspection, it seems that pytorch wasnt installed with CUDA. All good now!