How to limit GPU usage?

So I’d like to use these per_process_gpu_memory_fraction option from tensorflow, is it possible to add this (and other possible TF options) when using I’m not using Open-NMT like a library?

There is currently no way to set these flags except by modifying the code or using it as a library. I will look to add a way to set any tf.ConfigProto options from the command line.

1 Like

Thank you!



Wow! Thank you so much