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之大话题小函数(1)
Oct 10 Python
Python实现的简单算术游戏实例
May 26 Python
Python中threading模块join函数用法实例分析
Jun 04 Python
Python自动化测试Eclipse+Pydev 搭建开发环境
Aug 15 Python
Python测试人员需要掌握的知识
Feb 08 Python
解决tensorflow1.x版本加载saver.restore目录报错的问题
Jul 26 Python
Python实现简单的用户交互方法详解
Sep 25 Python
python实现ip地址查询经纬度定位详解
Aug 30 Python
Python进程池Pool应用实例分析
Nov 27 Python
python cv2截取不规则区域图片实例
Dec 21 Python
在Sublime Editor中配置Python环境的详细教程
May 03 Python
Scrapy基于scrapy_redis实现分布式爬虫部署的示例
Sep 29 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
header()函数使用说明
2006/11/23 PHP
php数组随机排序实现方法
2015/06/13 PHP
php生成静态html页面的方法(2种方法)
2015/09/14 PHP
Centos7安装swoole扩展操作示例
2020/03/26 PHP
利用javascript移动div层-javascript 拖动层
2009/03/22 Javascript
JS正则中的RegExp对象对象
2012/11/07 Javascript
在浏览器窗口上添加遮罩层的方法
2012/11/12 Javascript
JavaScript实现自己的DOM选择器原理及代码
2013/03/04 Javascript
extjs关于treePanel+chekBox全部选中以及清空选中问题探讨
2013/04/02 Javascript
JS 模态对话框和非模态对话框操作技巧汇总
2013/04/15 Javascript
子页向父页传值示例
2013/11/27 Javascript
Javascript MVC框架Backbone.js详解
2014/09/18 Javascript
JavaScript中switch语句的用法详解
2015/06/03 Javascript
AngularJS身份验证的方法
2016/02/17 Javascript
AngularJS入门教程引导程序
2016/08/18 Javascript
深入理解vue中slot与slot-scope的具体使用
2018/01/26 Javascript
vue-cli配置环境变量的方法
2018/07/09 Javascript
小程序实现分类页
2019/07/12 Javascript
微信小程序实现点击生成随机验证码
2020/09/09 Javascript
原生js实现九宫格拖拽换位
2021/01/26 Javascript
详解Python中表达式i += x与i = i + x是否等价
2017/02/08 Python
python3实现公众号每日定时发送日报和图片
2018/02/24 Python
解决python3 Pycharm上连接数据库时报错的问题
2018/12/03 Python
Python3 Click模块的使用方法详解
2020/02/12 Python
使用python-cv2实现视频的分解与合成的示例代码
2020/10/26 Python
专科应届生求职信
2013/11/24 职场文书
各营销点岗位职责范本
2014/03/05 职场文书
企业演讲稿范文大全
2014/05/20 职场文书
娱乐节目策划方案
2014/06/10 职场文书
工作检讨书500字
2014/10/19 职场文书
2015年医院后勤工作总结
2015/05/20 职场文书
《领导干部从政道德启示录》学习心得体会
2016/01/20 职场文书
《比的意义》教学反思
2016/02/18 职场文书
golang interface判断为空nil的实现代码
2021/04/24 Golang
超详细Python解释器新手安装教程
2021/05/10 Python
Springboot如何同时装配两个相同类型数据库
2021/11/17 Java/Android