Tensorflow限制CPU个数实例


Posted in Python onFebruary 06, 2020

安装

这里使用 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实现合并两个数组的方法
May 16 Python
Python中的推导式使用详解
Jun 03 Python
python 网络爬虫初级实现代码
Feb 27 Python
Python操作Excel之xlsx文件
Mar 24 Python
Python排序算法之选择排序定义与用法示例
Apr 29 Python
Python 二叉树的层序建立与三种遍历实现详解
Jul 29 Python
安装2019Pycharm最新版本的教程详解
Oct 22 Python
selenium+Chrome滑动验证码破解二(某某网站)
Dec 17 Python
python-OpenCV 实现将数组转换成灰度图和彩图
Jan 09 Python
Pytorch 使用opnecv读入图像由HWC转为BCHW格式方式
Jun 02 Python
Python 串口通信的实现
Sep 29 Python
Python竟然能剪辑视频
May 25 Python
如何解决tensorflow恢复模型的特定值时出错
Feb 06 #Python
python range实例用法分享
Feb 06 #Python
tensorflow 报错unitialized value的解决方法
Feb 06 #Python
TensorFlow实现保存训练模型为pd文件并恢复
Feb 06 #Python
解决TensorFlow模型恢复报错的问题
Feb 06 #Python
TensorFlow加载模型时出错的解决方式
Feb 06 #Python
基于Python实现人脸自动戴口罩系统
Feb 06 #Python
You might like
PHP防注入安全代码
2008/04/09 PHP
ThinkPHP自动转义存储富文本编辑器内容导致读取出错的解决方法
2014/08/08 PHP
PHP实现的基于单向链表解决约瑟夫环问题示例
2017/09/30 PHP
php中的依赖注入实例详解
2019/08/14 PHP
Laravel修改验证提示信息为中文的示例
2019/10/23 PHP
JS 实现BASE64_ENCODE和BASE64_DECODE(实例代码)
2013/11/13 Javascript
javascript数组操作总结和属性、方法介绍
2014/04/05 Javascript
js unicode 编码解析关于数据转换为中文的两种方法
2014/04/21 Javascript
jQuery $命名冲突解决方案汇总
2014/11/13 Javascript
JavaScript 学习笔记之操作符(续)
2015/01/14 Javascript
jQuery+css实现的tab切换标签(兼容各浏览器)
2016/01/28 Javascript
React入门教程之Hello World以及环境搭建详解
2017/07/11 Javascript
Angular4表单验证代码详解
2017/09/03 Javascript
vue对storejs获取的数据进行处理时遇到的几种问题小结
2018/03/20 Javascript
谈谈React中的Render Props模式
2018/12/06 Javascript
仿ElementUI实现一个Form表单的实现代码
2019/04/23 Javascript
layui表格内容溢出的解决方法
2019/09/06 Javascript
vue setInterval 定时器失效的解决方式
2020/07/30 Javascript
js实现炫酷光感效果
2020/09/05 Javascript
[51:20]完美世界DOTA2联赛PWL S2 Magma vs PXG 第一场 11.28
2020/12/01 DOTA
python中的装饰器详解
2015/04/13 Python
Pandas中DataFrame的分组/分割/合并的实现
2019/07/16 Python
关于pytorch中网络loss传播和参数更新的理解
2019/08/20 Python
Django使用中间件解决前后端同源策略问题
2019/09/02 Python
python实现将json多行数据传入到mysql中使用
2019/12/31 Python
css3中仿放大镜效果的几种方式原理解析
2020/12/03 HTML / CSS
C++:局部变量能否和全局变量重名
2014/03/03 面试题
学年自我鉴定
2014/01/16 职场文书
优秀部门获奖感言
2014/02/14 职场文书
欢迎标语大全
2014/06/21 职场文书
文明社区申报材料
2014/08/21 职场文书
公安局负责人查摆问题及整改方案
2014/09/27 职场文书
群众路线党员个人剖析材料
2014/10/08 职场文书
退税申请报告怎么写
2015/05/18 职场文书
毕业欢送晚会主持词
2019/06/25 职场文书
详解Golang如何优雅的终止一个服务
2022/03/21 Golang