刚自己解决了问题，给想用单卡GPU训练的人 single GPU used train TF keras CUDA cuDNN version

[ganchengguang]

提问时请尽可能提供如下信息：

基本信息

• 你使用的操作系统: win10
• 你使用的Python版本: python 3.6
• 你使用的Tensorflow版本: tensorflow-gpu==1.14.0
• 你使用的Keras版本: keras==2.3.1
• 你使用的bert4keras版本: 最新版本即可
• 你使用纯keras还是tf.keras: pure keras
• 你加载的预训练模型:google bert base
• 你使用的CUDA版本:10.0 https://developer.nvidia.com/cuda-10.0-download-archive
• • 你使用的cuDNN版本:cudnn-10.0-windows10-x64-v7.4.1.5

因为苏神没有写清楚版本对应情况，我经过自己几天的尝试了多个版本匹配，现在把我最后成功用GPU训练，bert4keras环境配置的各版本号写在这里，希望后面的人少走弯路，最后谢谢苏神提供的优化版keras-bert，还写了这么多例子。

[bojone]

tf版本和keras版本我都写清楚了吧，cuda和cudnn根据tf版本进行选择不就行了？还有啥弯路？

[qiujunhan]

cuda11可以吗，我那个服务器不能联网不想修改cuda版本，一直用不了