相关疑难解决方法(0)

如何减少 Tensorlfow/Keras 使用的 CPU 数量?

我正在使用 Tensorflow 2.0 的 Keras api。

在调用fit我的 Keras 模型时,它使用所有可用的 CPU。

我想限制使用的 CPU 的数量。然而,它曾经在以前版本的 Tensorflow 中工作的方式不能再使用了:

tf.keras.backend.set_session(tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(
       intra_op_parallelism_threads=2, inter_op_parallelism_threads=2)))
Run Code Online (Sandbox Code Playgroud)

AttributeError:模块“tensorflow.python.keras.api._v2.keras.backend”没有属性“set_session”

我怎么能那样做?

python keras tensorflow

8
推荐指数
1
解决办法
3650
查看次数

标签 统计

keras ×1

python ×1

tensorflow ×1