TensorFlow.jl icon indicating copy to clipboard operation
TensorFlow.jl copied to clipboard

Change number of cpu used

Open MaximeBouton opened this issue 6 years ago • 0 comments

Hello,

How can we control the number of cpu used by tensorflow?

I would like to achieve something similar as https://www.tensorflow.org/performance/performance_guide#optimizing_for_cpu

I noticed that config can be passed in tf.Session(), but I was not able to access the ConfigProto constructor. I tried:

julia> using TensorFlow; const tf=TensorFlow
TensorFlow

julia> config = tf.ConfigProto()
ERROR: UndefVarError: ConfigProto not defined

julia> config = tf.ProtoBuf.ConfigProto()
ERROR: UndefVarError: ConfigProto not defined

What is the correct way to define the configuration and the number of threads used? Thanks

MaximeBouton avatar Mar 06 '18 20:03 MaximeBouton