Skip to content
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.

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

Too much memory required on GPUs #1

Open
vvigilante opened this issue May 30, 2019 · 0 comments
Open

Too much memory required on GPUs #1

vvigilante opened this issue May 30, 2019 · 0 comments

Comments

@vvigilante
Copy link

vvigilante commented May 30, 2019

Hello,
the software goes out of memory when run on common GPUs. The problem arises when the session is run and does not occur when weights are loaded in detect.py or saved in convert_weights,py.

The problem is the same for yolov3 and yolov3-tiny, on different gpus.

It runs fine on CPU and takes little memory ( a few hundred megabytes).

When running yolov3-tiny on a Titan X GPU with allow_growth option, a few hundred megabytes are used during loading of weights but memory usage grows to 8GB (out of 12) when sess.run is called.

I think it is an anomalous behaviour.

GPU is a MX150, with 2GB memory, but I tried also others with same amount of memory.
Python 3.7 on windows, with Tensorflow 1.13.1, Cuda 10.1, Driver 425

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant