使用tensorflow2.3通过model.predictCUDA_ERROR_NOT_INITIALIZED

编程问答 2022-08-12

问题描述

我使用带有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）不起作用，并且发生以下错误。这个错误是什么意思？

然后我使用了docker image tensorflow/tensorflow:2.3.0-gpu-jupyter，并检查了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 ***

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

keraskeraspythontensorflow2.x