如何在一个内核上运行Tensorflow?

jojo123456

我在集群上使用Tensorflow,我想告诉Tensorflow仅在一个内核上运行(即使有更多可用的内核)。

有人知道这是否可能吗?

弗兰克·德农库尔特(Franck Dernoncourt)

要在一个CPU线程上运行Tensorflow,我使用:

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

device_count 限制使用的CPU数量,而不是内核或线程的数量。

tensorflow/tensorflow/core/protobuf/config.proto 说:

message ConfigProto {
  // Map from device type name (e.g., "CPU" or "GPU" ) to maximum
  // number of devices of that type to use.  If a particular device
  // type is not found in the map, the system picks an appropriate
  // number.
  map<string, int32> device_count = 1;

在Linux上,您可以运行sudo dmidecode -t 4 | egrep -i "Designation|Intel|core|thread"以查看有多少个CPU /核/线程,例如,以下有2个CPU,每个核有8个核,每个核有2个线程,总共2 * 8 * 2 = 32线程:

fra@s:~$ sudo dmidecode -t 4 | egrep -i "Designation|Intel|core|thread"
    Socket Designation: CPU1
    Manufacturer: Intel
            HTT (Multi-threading)
    Version: Intel(R) Xeon(R) CPU E5-2667 v4 @ 3.20GHz
    Core Count: 8
    Core Enabled: 8
    Thread Count: 16
            Multi-Core
            Hardware Thread
    Socket Designation: CPU2
    Manufacturer: Intel
            HTT (Multi-threading)
    Version: Intel(R) Xeon(R) CPU E5-2667 v4 @ 3.20GHz
    Core Count: 8
    Core Enabled: 8
    Thread Count: 16
            Multi-Core
            Hardware Thread

在Tensorflow 0.12.1和1.0.0以及Ubuntu 14.04.5 LTS x64和Ubuntu 16.04 LTS x64上进行了测试。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

C#任务是否在一个内核上运行?

来自分类Dev

如何在运行时在Tensorflow中查看或保存一个张量?

来自分类Dev

如何在同一个Ubuntu上运行两个Skype帐户?

来自分类Dev

如何在一台机器上运行一个外壳的多个版本

来自分类Dev

如何在Azure DevOps上一次运行一个发行版?

来自分类Dev

如何在一台机器上运行一个外壳的多个版本

来自分类Dev

内核版本如何在 Ubuntu 上运行?

来自分类Dev

CUDA如何在一个内核函数中启动新的内核调用?

来自分类Dev

如何在我的下一个内核升级中保留特定的旧内核?

来自分类Dev

另一个线程的代码如何在主线程上运行?

来自分类Dev

如何在另一个模块上添加运行时依赖项?

来自分类Dev

如何在一个简单的Java Web项目上运行声纳?

来自分类Dev

如何在一个EC2实例上运行多个应用程序?

来自分类Dev

如何在一个查询上运行这两行

来自分类Dev

如何在一个mesos从节点上运行多个docker容器?

来自分类Dev

如何在另一个模块上添加运行时依赖项?

来自分类Dev

如何在cpanel上运行一个简单的java程序

来自分类Dev

所有进程运行在同一个内核上

来自分类Dev

如何在不同的内核上运行4个线程(并行性)?

来自分类Dev

如何仅使用一个CPU内核运行程序?

来自分类Dev

如何在内核的一个数组中连接四个整数?

来自分类Dev

如何在一个内核模块中包含三个 IRQ 处理程序?

来自分类Dev

如何在东芝P70-A笔记本电脑上启用所有四个内核?(现在仅启用一个)

来自分类Dev

我如何在Windows 10 PC上创建一个Docker容器以在Raspberry Pi 4上运行

来自分类Dev

如何在另一个分区上运行Windows 7的第二个启动实例?

来自分类Dev

如何在一台机器上运行多台Cassandra服务器并形成一个集群环?

来自分类Dev

如何在多个终端上运行一个命令?

来自分类Dev

如何在JavaScript中依次运行一个函数?

来自分类Dev

如何在python中多次运行一个函数

Related 相关文章

  1. 1

    C#任务是否在一个内核上运行?

  2. 2

    如何在运行时在Tensorflow中查看或保存一个张量?

  3. 3

    如何在同一个Ubuntu上运行两个Skype帐户?

  4. 4

    如何在一台机器上运行一个外壳的多个版本

  5. 5

    如何在Azure DevOps上一次运行一个发行版?

  6. 6

    如何在一台机器上运行一个外壳的多个版本

  7. 7

    内核版本如何在 Ubuntu 上运行?

  8. 8

    CUDA如何在一个内核函数中启动新的内核调用?

  9. 9

    如何在我的下一个内核升级中保留特定的旧内核?

  10. 10

    另一个线程的代码如何在主线程上运行?

  11. 11

    如何在另一个模块上添加运行时依赖项?

  12. 12

    如何在一个简单的Java Web项目上运行声纳?

  13. 13

    如何在一个EC2实例上运行多个应用程序?

  14. 14

    如何在一个查询上运行这两行

  15. 15

    如何在一个mesos从节点上运行多个docker容器?

  16. 16

    如何在另一个模块上添加运行时依赖项?

  17. 17

    如何在cpanel上运行一个简单的java程序

  18. 18

    所有进程运行在同一个内核上

  19. 19

    如何在不同的内核上运行4个线程(并行性)?

  20. 20

    如何仅使用一个CPU内核运行程序?

  21. 21

    如何在内核的一个数组中连接四个整数?

  22. 22

    如何在一个内核模块中包含三个 IRQ 处理程序?

  23. 23

    如何在东芝P70-A笔记本电脑上启用所有四个内核?(现在仅启用一个)

  24. 24

    我如何在Windows 10 PC上创建一个Docker容器以在Raspberry Pi 4上运行

  25. 25

    如何在另一个分区上运行Windows 7的第二个启动实例?

  26. 26

    如何在一台机器上运行多台Cassandra服务器并形成一个集群环?

  27. 27

    如何在多个终端上运行一个命令?

  28. 28

    如何在JavaScript中依次运行一个函数?

  29. 29

    如何在python中多次运行一个函数

热门标签

归档