Keras下GPU/CPU模式切换
来源:互联网 发布:电脑锣编程软件有什么 编辑:程序博客网 时间:2024/06/14 03:43
1 确保环境
确保已经正确安装了keras, tensorflow/theano, cuda
在MacOS下面安装CUDA请参考:
mac osx/linux下如何将keras运行在GPU上
use cuda with macos
Ubuntu下面安装CUDA请参考:
配置深度学习环境的最后一步
2 切换gpu
来自官方的介绍How do I use keras with gpu
If you are running on the TensorFlow backend, your code will automatically run on GPU if any available GPU is detected. If you are running on the Theano backend, you can use one of the following methods:
Method 1: use Theano flags.
THEANO_FLAGS=device=gpu,floatX=float32 Python my_keras_script.py
The name ‘gpu’ might have to be changed depending on your device’s identifier (e.g. gpu0, gpu1, etc).Method 2: set up your .theanorc: Instructions
sudo vim ~/.theanorc
add these content
[global]
device=gpu
floatX=float32Method 3: manually set theano.config.device, theano.config.floatX at the beginning of your code:
import theano
theano.config.device = ‘gpu’
theano.config.floatX = ‘float32’
使用下面这个脚本来验证是否启动GPU:
from theano import function, config, shared, sandbox import theano.tensor as T import numpy import time vlen = 10 * 30 * 768 # 10 x #cores x # threads per core iters = 1000 rng = numpy.random.RandomState(22) x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) f = function([], T.exp(x)) print(f.maker.fgraph.toposort()) t0 = time.time() for i in xrange(iters): r = f() t1 = time.time() print("Looping %d times took %f seconds" % (iters, t1 - t0)) print("Result is %s" % (r,)) if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()]): print('Used the cpu') else: print('Used the gpu')
转自:
http://blog.csdn.NET/autoliuweijie/article/details/53157169
- Keras下GPU/CPU模式切换
- Keras下GPU/CPU模式切换
- Keras下GPU/CPU模式切换
- 在Keras中将cpu切换为gpu
- Keras 切换GPU
- Windows 10 安装 Cuda, Tensorflow, 和keras,附CPU/GPU切换小技巧
- Keras以及Tensorflow强制使用CPU,GPU
- windows系统下的keras(Theano/Tensorflow后端)CPU以及GPU加速环境搭建
- keras的GPU模式安装- theano 后端
- 170819 Anaconda两行命令安装tensorflow-gpu+keras-gpu及Gpu vs Cpu验证
- ubuntu下切换keras的后端
- 六、Ubuntu16.04下CPU或GPU模式caffe安装教程
- ubuntu 下GPU版的keras的环境搭建
- Anaconda下安装keras 配置(windows版本 GPU版)
- Win10 64bit下安装GPU版Tensorflow+Keras
- win10下配置GPU加速的Keras框架
- Win10 64bit下安装GPU版Tensorflow+Keras
- windows下安装gpu版本的keras\tensorflow
- 类继承
- lca(HDU 4718,The LCIS on the Tree)
- 动态规划,最长回文子序列(java实现),输入character,算法应该返回carac
- Ubuntu16.04下重启mysql和apach2
- CSS3 边框
- Keras下GPU/CPU模式切换
- linux编译openssl
- linux环境下的gcc
- web.xml 中spring-servlet.xml 和 application.xml 配置位置及含义
- 两个没有相互依赖的moudle之间的跳转
- 深入理解Linux修改hostname
- 授权+取消授权
- ViewPager适配器学习记录( pageAdapter和FragmentPagerAdapter/FragmentStatePagerAdapter))
- Thymeleaf模板引擎基本语法