时间戳

首页 > 解决方案 > 指定用于 Keras Tensorflow 模型推理的 CPU

python - 指定用于 Keras Tensorflow 模型推理的 CPU

问题描述

好吧。我知道我们可以通过以下方法限制 Keras（TF 后端）模型使用的内核数量：

 K.set_session(K.tf.Session(config=K.tf.ConfigProto(intra_op_parallelism_threads=2, inter_op_parallelism_threads=2,  device_count = {'CPU': 2})))

我们可以像这样指定单独的张量操作：

with tf.device('/cpu:0'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

但是，如果我们想指定一个供 Keras 模型使用的单个 CPU 的列表怎么办？

标签： pythontensorflowkeras

解决方案

我不认为你可以改变 Tensorflow 中的处理器亲和力，那是操作系统的水平。

但是，Linux 有一个有用的工具taskset可以帮助您。

例如，

taskset --cpu-list 0,1 python3 main.py

将核心 0 和核心 1 分配给运行的进程python3 main.py。

您可以使用htop.

推荐阅读