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在多玩图片上下载妹子图的实现代码
Aug 13 Python
Python使用urllib模块的urlopen超时问题解决方法
Nov 08 Python
在Python中使用zlib模块进行数据压缩的教程
Jun 26 Python
python如何使用unittest测试接口
Apr 04 Python
pandas apply 函数 实现多进程的示例讲解
Apr 20 Python
Python subprocess模块功能与常见用法实例详解
Jun 28 Python
python selenium 查找隐藏元素 自动播放视频功能
Jul 24 Python
python实现windows倒计时锁屏功能
Jul 30 Python
解决jupyter notebook打不开无反应 浏览器未启动的问题
Apr 10 Python
浅谈python3打包与拆包在函数的应用详解
May 02 Python
容易被忽略的Python内置类型
Sep 03 Python
PYTHON InceptionV3模型的复现详解
May 06 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之sphinx
2013/05/15 PHP
PHP获取数组中重复最多的元素的实现方法
2014/11/11 PHP
thinkPHP5框架分页样式类完整示例
2018/09/01 PHP
PHP配合fiddler抓包抓取微信指数小程序数据的实现方法分析
2020/01/02 PHP
firefox firebug中文入门教程 脚本之家新年特别版
2010/01/02 Javascript
JS中图片缓冲loading技术的实例代码
2013/08/29 Javascript
详解JavaScript基本类型和引用类型
2015/12/09 Javascript
原生js的数组除重复简单实例
2016/05/24 Javascript
jquery滚动条插件(可以自定义)
2016/12/11 Javascript
浅谈Angular的$q, defer, promise
2016/12/20 Javascript
Angular动态添加、删除输入框并计算值实例代码
2017/03/29 Javascript
ES6(ECMAScript 6)新特性之模板字符串用法分析
2017/04/01 Javascript
react-router JS 控制路由跳转实例
2017/06/15 Javascript
AngularJS实现页面跳转后自动弹出对话框实例代码
2017/08/02 Javascript
Javascript中从学习bind到实现bind的过程
2018/01/05 Javascript
js+css实现打字效果
2020/06/24 Javascript
Vue中的无限加载vue-infinite-loading的方法
2018/04/08 Javascript
NodeJS安装图文教程
2018/04/19 NodeJs
vue动态删除从数据库倒入列表的某一条方法
2018/09/29 Javascript
通过js给网页加上水印背景实例
2019/06/17 Javascript
vue如何使用async、await实现同步请求
2019/12/09 Javascript
python数据结构之图深度优先和广度优先实例详解
2015/07/08 Python
python  logging日志打印过程解析
2019/10/22 Python
米兰必去买手店排行榜首位:Antonioli
2016/09/11 全球购物
进修护士自我鉴定
2013/10/14 职场文书
求职者简历中的自我评价
2013/10/20 职场文书
四年级下册教学反思
2014/02/01 职场文书
2014年元旦促销活动方案
2014/02/22 职场文书
公司离职证明标准样本
2014/10/05 职场文书
小学生作文批改评语
2014/12/25 职场文书
黄石寨导游词
2015/02/05 职场文书
运动与健康自我评价
2015/03/09 职场文书
学校德育工作总结2015
2015/05/11 职场文书
2016应届毕业生自荐信范文
2016/01/28 职场文书
告诉你一个秘密:富人致富的五大优点
2019/07/11 职场文书
Jedis操作Redis实现模拟验证码发送功能
2021/09/25 Redis