search

opencv / open_vision_capsules

A set of libraries for encapsulating smart vision algorithms
Other
130 stars 48 forks source link

Remove allow_growth=True so applications can configure this instead #8

Closed velovix closed 4 years ago

velovix commented 4 years ago

Right now we set GPU options for all GPUs to allow memory growth, and there isn't a way for applications to change this behavior. We should instead not touch the GPU options and let applications configure this value by setting the TF_FORCE_GPU_ALLOW_GROWTH environment variable.

Nikhil1O1 commented 4 years ago

@velovix can I work on this,I'm new but willing to contribute

velovix commented 4 years ago

@Nikhil1O1 That would be great! The change should be relatively simple. All you'd have to do is remove allow_growth=True from here and add some documentation mentioning the TF_FORCE_GPU_ALLOW_GROWTH environment variable and linking to the Tensorflow docs on the subject. Maybe that documentation should be on this page. Let me know if you have any questions!

Nikhil1O1 commented 4 years ago

Awesome , I will see what i can do

Nikhil1O1 commented 4 years ago

24 @velovix I did the necessary changes, Please review it and let me know if you have any suggestions

Nikhil1O1 commented 4 years ago

@velovix If this one's solved in #24, please close this issue.