TensorFlow.jl
TensorFlow.jl copied to clipboard
Change number of cpu used
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