在trainer.py 中

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

下添加：

session_config.gpu_options.allow_growth = True

即可

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。