abhishekkrthakur/diffuzers

Host with GPU access

Opened this issue · 1 comments

Thank you for making it public, it's pretty awesome.
I have however a question that do you have any suggestion about the host that I can have the access to GPU?
Thanks,
Ai

Ik it's very late, but I just had the same problem and figured out this command does it: diffuzers app --device cuda. Keep in mind, that you need to have the correct torch version installed though. The normal one does NOT function for cuda. This makes it quite easy to get correct version.
Currently this is propably smart to use: pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118