Training Diffuser Model on Colab GPU - 'nvidia-smi' Error & Feasibility
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!
2024-08-23T07:44:45Z
hi@ManansiWhich notebook do you follow? Did you modify the model, dataset or code?I just triedDiffusion Models from Scratch - Hugging Face Diffusion Course. Even thought UNet2DModel takes more time, it seems Colab GPU can handle it.