CUDA_ERROR_NOT_INITIALIZED by model.predict() 使用 tensorflow2.3

Question

我将高效网络与 tensorflow2.3 API（keras==2.4.3）一起使用 https://www.tensorflow.org/api_docs/python/tf/keras/applications/efficientnet

我可以在 jupyterlab 上进行训练和预测。另一方面，在 Flask 实现时，模型检查点可以加载但 model.predict(numpy_img_array) 不起作用，并且发生了以下错误。这个错误是什么意思 ？

并且我使用tensorflow/tensorflow:2.3.0-gpu-jupyter了 docker图像，并且我检查了 model.summary() 工作正常。

真挚地

 2020-09-04 11:22:40.559654: E tensorflow/stream_executor/cuda/cuda_driver.cc:951] could not synchronize on CUDA context: CUDA_ERROR_NOT_INITIALIZED: initialization error :: *** Begin stack trace ***
   tensorflow::CurrentStackTrace()
   stream_executor::gpu::GpuDriver::SynchronizeContext(stream_executor::gpu::GpuContext*)
   stream_executor::StreamExecutor::SynchronizeAllActivity()
   tensorflow::GPUUtil::SyncAll(tensorflow::Device*)
   tensorflow::BaseGPUDevice::Sync()
   tensorflow::TensorHandle::CopyToDevice(tensorflow::EagerContext const&, tensorflow::Device*, tensorflow::Tensor*)
   tensorflow::TensorHandle::Resolve(tensorflow::Status*)
   TFE_TensorHandleResolve
   _PyEval_EvalFrameDefault
   _PyFunction_FastCallDict
   _PyObject_FastCallKeywords
   _PyEval_EvalFrameDefault
   PyObject_Call
   _PyFunction_FastCallDict
   _PyObject_FastCallKeywords
   _PyEval_EvalFrameDefault
   PyEval_EvalCode
   PyRun_FileExFlags
   PyRun_SimpleFileExFlags
   Py_Main
   main
   __libc_start_main
   _start
 *** End stack trace ***