yuval-alaluf / stylegan3-editing

Official Implementation of "Third Time's the Charm? Image and Video Editing with StyleGAN3" (AIM ECCVW 2022) https://arxiv.org/abs/2201.13433

Home Page:https://yuval-alaluf.github.io/stylegan3-editing/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can I run the repo without GPU?

zydjohnHotmail opened this issue · comments

Hello:
I want to use the repo to generate some images. However, my PC has NVIDIA Geforce gt 1030 graphic card. This graphic card has no ability to run any CUDA operation, but installing CUDA package is OK. But my PC has 32-core CPU with 256GB RAM, so I think CPU is powerful enough to do almost all the jobs.
I don't want to train any model, I just want to use pre-trained (.PT) models to generate some human faces or animal images.
Let me know in which python file, I can change so that the image predition use only CPU not GPU.
Thanks,

you can start with a 3060 12gb
https://github.com/NVlabs/stylegan3#requirements
#12
(CPU may be possible with some modifications, but is not inherently supported)
https://stackoverflow.com/questions/51730880/where-do-i-get-a-cpu-only-version-of-pytorch
You will need to create a fork from this repo

commented

CPU may be possible with some modifications, but is not inherently supported.

Reference: https://github.com/yuval-alaluf/stylegan3-editing#prerequisites