脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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中模块string.py详解

    Python中模块string.py详解

    这篇文章主要介绍了Python中模块之string.py的相关资料，文中介绍的非常详细，对大家具有一定的参考价值，需要的朋友们下面来一起看看吧。
    2017-03-03
  • 详解Python的三种拷贝方式

    详解Python的三种拷贝方式

    Python中有三种拷贝方式分别是浅拷贝、深拷贝和赋值拷贝，这篇文章通过实例代码给大家介绍了Python的三种拷贝方式,需要的朋友可以参考下
    2020-02-02
  • Django实现drf搜索过滤和排序过滤

    Django实现drf搜索过滤和排序过滤

    当我们需要对后台的数据进行过滤的时候，drf有两种，搜索过滤和排序过滤。本文就详细的介绍这两种的实现，感兴趣的可以了解一下
    2021-06-06
  • Python3监控疫情的完整代码

    Python3监控疫情的完整代码

    这篇文章主要介绍了Python3监控疫情的完整代码，代码简单易懂，非常不错具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-02-02
  • 基于plt.title无法显示中文的快速解决

    基于plt.title无法显示中文的快速解决

    这篇文章主要介绍了基于plt.title无法显示中文的快速解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-05-05
  • Python中强大的函数map filter reduce使用详解

    Python中强大的函数map filter reduce使用详解

    Python是一门功能丰富的编程语言,提供了许多内置函数,以简化各种编程任务,在Python中,map(),filter()和reduce()是一组非常有用的函数,它们允许对可迭代对象进行操作,从而实现数据转换、筛选和累积等操作,本文将详细介绍这三个函数,包括它们的基本用法和示例代码
    2023-11-11
  • Python加密与解密模块hashlib与hmac

    Python加密与解密模块hashlib与hmac

    这篇文章介绍了Python中的加密与解密模块hashlib与hmac，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2022-06-06
  • python 类对象的析构释放代码演示

    python 类对象的析构释放代码演示

    这篇文章主要介绍了python 类对象的析构释放代码演示，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的朋友可以参考一下
    2022-09-09
  • python 打印dict的key与value方式

    python 打印dict的key与value方式

    这篇文章主要介绍了python 打印dict的key与value方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2022-11-11
  • Python中选择排序的实现与优化

    Python中选择排序的实现与优化

    选择排序（Selection Sort）是一种简单但有效的排序算法，本文将详细介绍选择排序算法的原理和实现，并提供相关的Python代码示例，需要的可以参考一下
    2023-06-06

最新评论