How to limit GPU usage?

tensorflow

(Xun468) #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?


(Guillaume Klein) #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.


(Xun468) #3

Thank you!


(Guillaume Klein) #4

FYI:


(Xun468) #5

Wow! Thank you so much