使用 keras 进行代码时，tensorflow-gpu 卡在 epoch1

Question

突然使用keras执行代码时，出现如下警告：

An error ocurred while starting the kernel
2019???????? 16:08:03.433226: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1405] Found device 0 with properties:
name: GeForce RTX 2060 major: 7 minor: 5 memoryClockRate(GHz): 1.2
pciBusID: 0000:01:00.0
totalMemory: 6.00GiB freeMemory: 4.89GiB
2019???????? 16:08:03.435266: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1484] Adding visible gpu devices: 0
2019???????? 16:08:05.728803: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:
2019???????? 16:08:05.730056: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:971] 0
2019???????? 16:08:05.730321: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:984] 0: N
2019???????? 16:08:05.730723: I c:\users\user\source\repos\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1097] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4644 MB memory) ‑> physical GPU (device: 0, name: GeForce RTX 2060, pci bus id: 0000:01:00.0, compute capability: 7.5

我的系统信息：

• win10
• cuda9.2
• cdunn7.2
• python3.65
• TensorFlow 版本：1.10.0
• GPU/CPU:RTX2060,i7 9750h
• 内存：16g

但是当我直接使用 tensorflow 时，没有通过 keras 处理代码，没有任何警告并且完全成功。 尽管如此，我仍然更喜欢 keras 而不是直接使用 tensorflow。

Answer 1

对我来说，此设置适用于 RTX2060 和 i7 9750h：

win10，

CUDA 10.0，

cuDNN 7.4，

python 3.68，

tensorflow_gpu-1.13.1

参考资料： https://www.tensorflow.org/install/source#linux