Training Diffuser Model on Colab GPU - 'nvidia-smi' Error & Feasibility

Beginners
1

Hi everyone,

I’m trying to train a diffuser model for image generation using the Diffusers library, following the code from Hugging Face. However, I’m encountering an error when trying to use the Colab GPU:
FileNotFoundError: [Errno 2] No such file or directory: ‘nvidia-smi’
What might be causing this error? Is it related to the GPU setup on Colab, or is there something wrong with my environment configuration?
Also, is it actually feasible to train a diffuser model on Colab, given its GPU limitations? Any advice on the best way to approach this would be greatly appreciated!

2

hi @Manansi
Which notebook do you follow? Did you modify the model, dataset or code?

I just tried Diffusion Models from Scratch - Hugging Face Diffusion Course. Even thought UNet2DModel takes more time, it seems Colab GPU can handle it.