anishathalye / neural-style

Neural style in TensorFlow! 🎨
https://anishathalye.com/an-ai-that-can-mimic-any-artist/
GNU General Public License v3.0
5.54k stars 1.52k forks source link

Allocator (GPU_0_bfc) ran out of memory trying to allocate XYGiB with freed_by_count=0. #162

Closed peakpwr closed 4 years ago

peakpwr commented 4 years ago

Hello, I'm facing some problems as my GPU is running out of memory I guess. When I try to run the script on a high resolution content (>3000x2500px) the script interrupts with the error: Allocator (GPU_0_bfc) ran out of memory trying to allocate XYGiB with freed_by_count=0. My GPU has 11GB of RAM.

How can the script be changed so these high res pictures can be used? Or is it only possible if I use a GPU with more RAM?

anishathalye commented 4 years ago

See #91 and #135