keras使用gpu训练(tensor1.15 + cuda10.2)

模型搭建

首先写一个简单的 训练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

你可能感兴趣的:(图像处理)