Host with GPU access
Opened this issue · 1 comments
vuhongai commented
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
Luh-code commented
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