当前位置：网站首页>限制tensorflow使用Cpu核数

限制tensorflow使用Cpu核数

2022-08-04 21:52:00 【天天进步2015】

一开始，用tensorflow训练时，使用默认设置，运行时Cpu占用率过高，机器声音很响。

后来，改用下面的代码限制tensorflow使用Cpu核数。

session_conf = tf.ConfigProto(
      intra_op_parallelism_threads=1,
      inter_op_parallelism_threads=1)
sess = tf.Session(config=session_conf)

原网站

版权声明
本文为[天天进步2015]所创，转载请带上原文链接，感谢
https://blog.csdn.net/exlink2012/article/details/126141132

边栏推荐

猜你喜欢

随机推荐