a312863063 / seeprettyface-generator-wanghong

这是一个用StyleGAN训练出的网红脸生成器

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Ran Out of Memory

YukihimeX opened this issue · comments

W tensorflow/core/common_runtime/bfc_allocator.cc:211] Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.14GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.

Hey man, really love your result and I was trying to run this repository but I kept getting this out of memory... any idea how I go about fixing this?

I build is:

  • Windows 10
  • GTX 970 and 1070 (separate PC respectively)
  • Python 3.6.9
  • Tensorflow-gpu 1.14
  • VS2017&2019 Community
  • CUDA 10.0 + cuDNN 7.6.5

For some reason, my PC with the 1070 Graphic Card always freezes instead of showing me the message above. At least my GTX 970 PC show me the message above.

Stylegan