Keras以及Tensorflow强制使用CPU
来源:互联网 发布:淘宝卖什么吃的赚钱 编辑:程序博客网 时间:2024/05/17 07:20
Keras如果是使用Theano后端的话,应该是自动不使用GPU只是用CPU的,启动GPU使用Theano内部命令即可。
对于Tensorflow后端的Keras以及Tensorflow会自动使用可见的GPU,而我需要其必须只运行在CPU上。网上查到三种方法,最后一种方法对我有用,但也对三种都做如下记录:
- 使用tensorflow的
with tf.device('/cpu:0'):
函数。简单操作就是把所有命令都放在前面所述的域里面。 - 使用tensorflow声明Session时的参数:
关于tensorflow中Session中的部分参数设置,以及Keras如何设置其调用的Tensorflow的Session,可以参见Keras设定GPU使用内存大小(Tensorflow backend)。
对于Tensorflow,声明Session的时候加入device_count={'gpu':0}
即可,代码如下:
import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0}))
对于Keras,则调用后端函数,设置其使用如上定义的Session即可,代码如下:
import tensorflow as tfimport keras.backend.tensorflow_backend as KTFKTF.set_session(tf.Session(config=tf.ConfigProto(device_count={'gpu':0})))
对于多线程以及GPU内存设置等可以参见Keras设定GPU使用内存大小(Tensorflow backend);更多详细内容请见tensorflow官网。
3 . 第三种是使用CUDA_VISIBLE_DEVICES命令行参数,代码如下:
CUDA_VISIBLE_DEVICES="" python3 train.py
阅读全文
1 0
- Keras以及Tensorflow强制使用CPU
- Keras以及Tensorflow强制使用CPU,GPU
- Keras使用tensorflow代码
- keras tensorboard的使用, 设置GPU使用的内存, 强制只使用cpu
- windows系统下的keras(Theano/Tensorflow后端)CPU以及GPU加速环境搭建
- keras tensorboard的使用, 设置指定GPU及其内存, 强制只使用cpu
- keras 与tensorflow 混合使用
- keras以及tensorflow中文文档地址
- Mac CPU版本tensorflow,Theano,tensorlayer,keras搭建
- win10+anaconda+tensorflow+keras环境搭建教程(CPU版)
- Keras + TensorFlow
- Keras设定GPU使用内存大小(Tensorflow backend)
- Keras设定GPU使用内存大小(Tensorflow backend)
- 使用Keras和Tensorflow检测恶意请求
- Tensorflow、Keras使用过程中的问题
- TensorFlow如何充分使用所有CPU核数,提高TensorFlow的CPU使用率,以及Intel的MKL加速
- keras模型导出到tensorflow,以及pb模型导出
- 基于tensorflow的keras中GPU的使用
- 第七次项目3实验
- 新网域名添加AAAA记录
- probelm D 大数+组合dp
- Android MarkDownView中显示gif图片
- JVM源码分析之一个Java进程究竟能创建多少线程
- Keras以及Tensorflow强制使用CPU
- ffmpeg实战教程(十二)为视频添加/更换背景音乐
- MYSQL索引优化: IN 和 OR 替换为 union all
- 运放参数及选型表
- 哈希表(亦称散列表)HashMap源码读后感,实现原理剖析
- Android Notification常见样式总结
- java 内存
- VIM退出命令
- Linux文件系统操作与磁盘管理