search

kwotsin / mimicry

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.
MIT License
602 stars 62 forks source link

INFO: Propagated batch xxxx/1000 is very slow #12

Closed Johnson-yue closed 4 years ago

Johnson-yue commented 4 years ago

Hi， this progress is very slow and maybe it only using CPU for computiing！！！ （2~3 sec/batch） can you update it for GPU using？

kwotsin commented 4 years ago

Hi @Johnson-yue, I believe simply installing the GPU version of TensorFlow should automatically use the GPU for the propagation. Perhaps you can try to use conda to install TensorFlow-GPU, e.g. conda install tensorflow-gpu

Hope it helps!

Johnson-yue commented 4 years ago

oh, this progress is depend on Tensorflow !! I got it