Only CPU usage on Windows

peterwilli / sd-leap-booster

Fast finetuning using a booster model that puts the initial state to a local minimum

MIT License

113 stars 12 forks source link

Only CPU usage on Windows #1

Closed ghost closed 1 year ago

ghost commented 1 year ago

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?

ghost commented 1 year ago

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