How to limit GPU usage?

Support
1

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?

2

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
3

Thank you!

4

FYI:

2 Likes
5

Wow! Thank you so much