Spaces:
Running
on
A10G
GPU not recognized when I try to run it locally
Hey, so I'm trying to run this locally and it's working - but not as expected. It recognizes my 3080 Ti but when I run it it only uses my CPU, which of course is terribly slow. Any way anyone knows how to fix that?
umm make sure you have pytorch installed with CUDA enabled. for example install with "pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117" sometimes if you just do pip install torch you get CPU only version. you can check with
python
import torch
torch.cuda.is_available()
a nice way to run locally now is to use automatic1111 stable diffusion web ui and the new CLIP Interrogator extension
https://github.com/pharmapsychotic/clip-interrogator-ext
Thanks pharma, that was exactly what fixed it! Highly appreciated <3