我正在运行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
有人对造成这种情况的原因有任何想法吗?