hollowstrawberry/kohya-colab

Falling back to cpu

Closed this issue · 4 comments

remote3513 commented

Hi,

I've noticed since yesterday i get a line in the script of the Colab page: "An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu."

I have the GPU enabled in the Colab page and i've also tried all of the other GPU's but i get the error message with every GPU.

However, I am able to train despite "falling back to the CPU".

I just want to know if this an error on my side or perhaps in the script?

Cheers!

ArmyOfPun1776 commented

I get the same message when training. It's definitely a collab thing as nothing run in collab is using anything on your local system. I'm not even sure if this an actual issue or intended behavior. Seeing that the trainer doesn't break because of it, I'm leaning toward intended behavior.

  • 👀1
hollowstrawberry commented

The training works fine despite this warning, and it's clearly using the gpu, so you can ignore it.

  • 🎉2
remote3513 commented

Ok! Cheers guys!

firofame commented

Screenshot 2024-06-01 at 10 15 19 PM

"An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu."