При использовании кераса для неожиданного продолжения кода came вышло предупреждение, которое выглядит следующим образом:
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
Но когда я использовал тензор потока напрямую, без обработки кода керасом, не было никаких предупреждений, и это было совершенно успешно.
Несмотря на все это, я все же предпочитаю керасы, а не напрямую использую тензорный поток.