New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

it's OOM if i use tensorflow-gpu. Did i do something wrong? #15

Open

JiayunLin opened this issue Dec 27, 2017 · 2 comments

JiayunLin commented Dec 27, 2017

I use 1060 with 3G vram, and it's useless.......
train_on_batch() use a lot of memory
Did i do something wrong?

Rnijland commented Jan 25, 2018

Same with GTX 960 4gb....

Rnijland commented Jan 25, 2018 •

edited

Loading

Fixed it, the model.py has a formula that allocates 8gb ram. Probably someone with a 1080gtx made it.

Cant paste the right code here, as star symbols won`t show... Follow This:

https://pastebin.com/7dyWECmT

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment