CUDNN_STATUS_ALLOC_FAILED в прямом запуске Pycharm, но не в консоли Python - PullRequest
0 голосов
/ 30 мая 2019

Я запускаю программу обнаружения объектов Tensorflow, но обнаружил очень странный случай. Когда я напрямую запускаю один и тот же контент в файле, Cuda всегда не работает.

Но если я копирую и вставляю контент в консоль Python, он работает без ошибок и выдает правильное обнаружение.

2019-05-30 15:38:36.569895: E tensorflow/stream_executor/cuda/cuda_dnn.cc:373] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
2019-05-30 15:38:36.570676: E tensorflow/stream_executor/cuda/cuda_dnn.cc:373] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

Кто-нибудь имеет представление о том, что может вызвать это?

...