安装GPU支持
Keras集成了TensorFlow,但是使用TensorFlow的GPU计算要安装安装 CUDA 和配置 NVIDIA cuDNN ,参照以下博客安装(看到安装顺序的第2步就可以了):
win10搭建tensorflow-gpu环境
注意:自行搜索自己显卡对应的CUDA,再搜索与CUDA对应的cudnn版本。
它安装的是CUDA9.0,而我安装的是 cuda_10.1.168_425.25_win10.exe,
与其对应的cuDNN版本是cudnn-10.1-windows10-x64-v7.6.0.64(这个需要登录才能下载)。
除了版本不同,其他操作都一样的。
以下是从上面博客的第3步开始的:,所以先完成上面配置先再进行以下操作。 这里是安装tensorflow-gpu版本的