@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

總結

以上是生活随笔為你收集整理的【小白学习keras教程】九、keras 使用GPU和Callbacks模型保存的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。