【小白学习keras教程】九、keras 使用GPU和Callbacks模型保存

@Author：Runsen

GPU

在gpu上训练使训练神经网络比在cpu上运行快得多

Keras支持使用Tensorflow和Theano后端对gpu进行培训

文档： https://keras.io/getting-started/faq/#how-can-i-run-keras-on-gpu

安装GPU

首先，下载并安装CUDA&CuDNN（假设您使用的是NVIDIA gpu）

安装url: https://developer.nvidia.com/cudnn

然后，通过在cmd或terminal中键入以下命令来安装tensorflow gpu（启用gpu的tensorflow版本）

pip install tensorflow gpu

然后检查机器正在使用GPU设备

在下面例子中，我有一个GPU设备（其名称为“/device:GPU:0”）

如果使用的是Google Colab，只需将运行时类型更改为“GPU”

import tensorflow as tf from tensorflow

1

GPU加速云服务器 Keras TensorFlow

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。