kwotsin / mimicry

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.

INFO: Propagated batch xxxx/1000 is very slow

Johnson-yue opened this issue 4 years ago · comments

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