在trainer.py 中

session_config = tf.ConfigProto(allow_soft_placement=True,log_device_placement=False)

下添加：

session_config.gpu_options.allow_growth = True

即可

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的object detection错误之Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。