脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Tensorflow限制CPU个数

Tensorflow限制CPU个数实例

 更新时间：2020年02月06日 15:00:50   作者：kongxx  
今天小编就为大家分享一篇Tensorflow限制CPU个数实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

安装

这里使用 Pip 来安装 Tensorflow CPU 版

$ sudo pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

安装完成后运行库中自带的手写识别例子来检查安装是否成功

$ cd /usr/lib/python2.7/site-packages/tensorflow/models/image/mnist
$ python convolutional.py
...

或者运行

$ python -m tensorflow.models.image.mnist.convolutional
...

限制CPU个数

对于上面用到的手写识别例子来说，需要修改文件 /usr/lib/python2.7/site-

packages/tensorflow/models/image/mnist/convolutional.py 中创建 Session 部分

修改前

  with tf.Session(config=config) as s:

修改后

  cpu_num = int(os.environ.get('CPU_NUM', 1))
  config = tf.ConfigProto(device_count={"CPU": cpu_num},
        inter_op_parallelism_threads = cpu_num,
        intra_op_parallelism_threads = cpu_num,
        log_device_placement=True)

  with tf.Session(config=config) as s:

修改完成后，使用环境变量 CPU_NUM 来指定需要使用的 CPU 个数，然后再次运行手写识别例子

$ export CPU_NUM=2
$ python -m tensorflow.models.image.mnist.convolutional

运行后，使用 top 命令来查看程序的 CPU 使用情况。

以上这篇Tensorflow限制CPU个数实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python的函数使用示例详解

    Python的函数使用示例详解

    在Python的函数中，我们将其分为内置函数、自定义函数、main函数三个模块，当然，使用的过程中会涉及到变量以及参数，这些都会举例进行说明，对Python函数使用相关知识感兴趣的朋友跟随小编一起看看吧
    2021-12-12
  • Kears 使用:通过回调函数保存最佳准确率下的模型操作

    Kears 使用:通过回调函数保存最佳准确率下的模型操作

    这篇文章主要介绍了Kears 使用:通过回调函数保存最佳准确率下的模型操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • Python实现将照片变成卡通图片的方法【基于opencv】

    Python实现将照片变成卡通图片的方法【基于opencv】

    这篇文章主要介绍了Python实现将照片变成卡通图片的方法,涉及Python基于opencv库进行图片处理的相关操作技巧,需要的朋友可以参考下
    2018-01-01
  • ubuntu17.4下为python和python3装上pip的方法

    ubuntu17.4下为python和python3装上pip的方法

    今天小编就为大家分享一篇ubuntu17.4下为python和python3装上pip的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-06-06
  • NumPy内存布局的实现

    NumPy内存布局的实现

    本文主要介绍了NumPy内存布局的实现,括连续内存布局（C顺序）和分散内存布局（Fortran顺序）,并通过实例演示如何操作数组的内存布局,感兴趣的可以了解一下
    2024-01-01
  • Queue队列中join()与task_done()的关系及说明

    Queue队列中join()与task_done()的关系及说明

    这篇文章主要介绍了Queue队列中join()与task_done()的关系及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2023-02-02
  • Python脚本读取Consul配置信息的方法示例

    Python脚本读取Consul配置信息的方法示例

    本文主要介绍了Python脚本读取Consul配置信息的方法示例，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-11-11
  • Pygame做一期吃豆子游戏的示例代码

    Pygame做一期吃豆子游戏的示例代码

    这篇文章主要介绍了Pygame做一期吃豆子游戏的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-03-03
  • python中强制关闭线程与协程与进程方法

    python中强制关闭线程与协程与进程方法

    python使用中多线程、多进程、多协程使用是比较常见的。那么如果在多线程等的使用，我们这个时候我们想从外部强制杀掉该线程请问如何操作？这篇文章带你介绍，感兴趣的同学可以参考阅读
    2023-03-03
  • 使用Pycharm(Python工具)新建项目及创建Python文件的教程

    使用Pycharm(Python工具)新建项目及创建Python文件的教程

    这篇文章主要介绍了使用Pycharm(Python工具)新建项目及创建Python文件的教程,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2020-04-04

最新评论