@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
 

文章来源: maoli.blog.csdn.net，作者：刘润森！，版权归原作者所有，如需转载，请联系作者。

原文链接：maoli.blog.csdn.net/article/details/118750940