search

Puzer / stylegan-encoder

StyleGAN Encoder - converts real images to latent space
Other
1.07k stars 166 forks source link

Question: running on CPU? #14

Closed jmiller656 closed 5 years ago

jmiller656 commented 5 years ago

Do you know if it's possible to run stylegan on a GPU machine? Been trying to run a version of stylegan I've trained on my desktop machine and would like to mess around with the checkpoints while it's training. Keep seeing that the operations are specifically set to use GPU still.

Specifically, this error: 


:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device.
         [[Node: Gs/_Run/Gs/latents_in = Identity[T=DT_FLOAT, _device="/device:GPU:0"](Gs/_Run/split)]]```
pbaylies commented 5 years ago

@jmiller656 the original StyleGAN code is designed to be run with at least one GPU; if you just want to play around, I'd suggest using Google Colab, where you can load up a free notebook in the cloud and use one of their GPUs. Otherwise, you'd need to write or find an implementation that doesn't require a GPU.

jmiller656 commented 5 years ago

Hmm, was trying to not have to drag my model onto google drive and deal with all that... guess I have no choice. Thanks!