模型搭建
首先寫一個簡單的 訓練cifar10 的模型,可參考
Keras入門課3 – 使用CNN識別cifar10數據集
CPU訓練
安裝pip install tensorflow==1.15 , 默認使用CPU 訓練, 一個epoch訓練時間爲:141s
GPU訓練
注意:tensorflow-gpu-cudnn-cuda對應版本
正確的步驟是按照對照表去安裝。 PS :tensorflow1.15 對應的也是cuda10.0
使用GPU 訓練, 一個epoch訓練時間爲:21s !!!
但是不幸的是,你提前安裝了cuda10.2 , 又深知重裝之痛苦。那就繼續往下看吧
Centos8 運行CUDA10.2+Tensorflow1.15.0