【小白学习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本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！