Keras可用配置记录

由于tensorflow、keras之类的玩意更新实在太快,网上扒的之前的轮子代码好多在新版的环境下根本运行不了。
所以为了跑一个WGAN-GP的轮子,花了一天的时间摸索用什么样的配置。
现记录如下:

CUDA 10.0
CUDNN 10.0 V7.4.1.5
python 3.6.12
tensorflow-gpu 1.14
keras 2.0.9

网上很多代码都是降级之后才能正常运行。
还是要读论文然后在新版环境下复现啊。光玩轮子可不行。

原文地址:https://www.cnblogs.com/soul-M/p/13885710.html