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 ORM框架SQLAlchemy学习笔记之数据添加和事务回滚介绍
Jun 10 Python
Python的加密模块md5、sha、crypt使用实例
Sep 28 Python
python3实现UDP协议的服务器和客户端
Jun 14 Python
rabbitmq(中间消息代理)在python中的使用详解
Dec 14 Python
TF-IDF与余弦相似性的应用(二) 找出相似文章
Dec 21 Python
解决pip install xxx报错SyntaxError: invalid syntax的问题
Nov 30 Python
Python3实现对列表按元组指定列进行排序的方法分析
Dec 22 Python
详解Python装饰器
Mar 25 Python
Python 批量刷博客园访问量脚本过程解析
Aug 30 Python
python tkinter控件布局项目实例
Nov 04 Python
Tensorflow:转置函数 transpose的使用详解
Feb 11 Python
Django 自定义404 500等错误页面的实现
Mar 08 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
一首老MP3,致敬WAR3经典
2021/03/08 魔兽争霸
Smarty变量调节器失效的解决办法
2014/08/20 PHP
smarty高级特性之对象的使用方法
2015/12/25 PHP
PHP获取当前系统时间的方法小结
2018/10/03 PHP
零基础php编程好学吗
2019/10/11 PHP
在js中使用"with"语句中跨frame的变量引用问题
2007/03/08 Javascript
javascript:FF/Chrome与IE动态加载元素的区别说明
2014/01/26 Javascript
JavaScript操作XML文件之XML读取方法
2015/06/09 Javascript
JS中使用变量保存arguments对象的方法
2016/06/03 Javascript
浅谈JavaScript中的this指针和引用知识
2016/08/05 Javascript
jquery dialog获取焦点的方法
2017/02/09 Javascript
jQuery 实现图片的依次加载图片功能
2017/07/06 jQuery
vue移动端路由切换实例分析
2018/05/14 Javascript
浅谈Webpack核心模块tapable解析
2018/09/11 Javascript
实例讲解v-if和v-show的区别
2019/01/31 Javascript
ES6 如何改变JS内置行为的代理与反射
2019/02/11 Javascript
微信小程序分享功能onShareAppMessage(options)用法分析
2019/04/24 Javascript
JQuery省市联动效果实现过程详解
2020/05/08 jQuery
js canvas实现五子棋小游戏
2021/01/22 Javascript
python实现的一个火车票转让信息采集器
2014/07/09 Python
Python中方法链的使用方法
2016/02/23 Python
使用PyInstaller将Python程序文件转换为可执行程序文件
2016/07/08 Python
python定时检测无响应进程并重启的实例代码
2019/04/22 Python
python 列表转为字典的两个小方法(小结)
2019/06/28 Python
Django中间件拦截未登录url实例详解
2019/09/03 Python
Django之使用内置函数和celery发邮件的方法示例
2019/09/16 Python
解决django的template中如果无法引用MEDIA_URL问题
2020/04/07 Python
python3代码输出嵌套式对象实例详解
2020/12/03 Python
财务部岗位职责
2013/11/19 职场文书
高中军训感言800字
2014/03/05 职场文书
车间主任岗位职责范本
2015/04/08 职场文书
小学运动会通讯稿
2015/07/18 职场文书
如何写好开幕词?
2019/06/24 职场文书
七个非常实用的Python工具包总结
2021/06/15 Python
html中显示特殊符号(附带特殊字符对应表)
2021/06/21 HTML / CSS
mybatis中注解与xml配置的对应关系和对比分析
2021/08/04 Java/Android