【小白學習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小時內刪除侵權內容。