Nvidia Only ? about draggan

[Zeqiang-Lai]

Yes, you need GPU. It might be possible to run it with CPU, but it would be extremely slow and I didn't try it.

[Leonetienne]

I get this error aswell.
I have an NVIDIA gpu.
Stable diffusion runs fine on it on this install.
torch.cuda.is_available() returns true

i am using ubuntu

[Zeqiang-Lai]

I get this error aswell. I have an NVIDIA gpu. Stable diffusion runs fine on it on this install. torch.cuda.is_available() returns true

i am using ubuntu

You need CUDA installed to compile the custom ops of stylegan2.

Please refer to https://developer.nvidia.com/cuda-downloads

[Myzel394]

@Zeqiang-Lai I currently do not have a Nvidia GPU but was thinking about buying one. Do you have any recommendations for GPUs that work very well with your DragGAN implementation?

[Zeqiang-Lai]

@Zeqiang-Lai I currently do not have a Nvidia GPU but was thinking about buying one. Do you have any recommendations for GPUs that work very well with your DragGAN implementation?

I would recommend Nvidia 3090/4090 with 24GB video memory. They works perfect for me.

This implementation need a lot video memory, 9 GB at least for 1024x1024 pictures btw.