Я пытаюсь использовать tensorflow-gpu для обнаружения своих пользовательских объектов. Я создал среду conda, загрузил и установил CUDA и CuDNN 10.0 с моей версией tensorflow-gpu 1.14. Однако я получил эту ошибку
ResourceExhaustedError: 2 root error(s) found.
(0) Resource exhausted: OOM when allocating tensor with shape[32,1792,1792,3,2] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator gpu_host_bfc
[[{{node Tile_5}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
(1) Resource exhausted: OOM when allocating tensor with shape[32,1792,1792,3,2] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator gpu_host_bfc
[[{{node Tile_5}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
[[training/Adam/cond_175/Switch_2/_6707]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
0 successful operations.
0 derived errors ignored.
Следуя этому веб-сайту Train Object Detection AI с 6 строками кода.
