在Pycharm中直接运行CUDNN_STATUS_ALLOC_FAILED,但在Python控制台中不运行

时间:2019-05-30 07:49:52

标签: tensorflow cudnn

我正在运行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

有人对造成这种情况的原因有任何想法吗?

0 个答案:

没有答案