https://github.com/tensorflow/models/issues/4881
问题二:
GPU训练对显存与内存使用是基于贪心算法,它会一直尝试获取更多内存,大概训练了100左右step就会爆出如下的错误:
tensorflow.python.framework.errors_impl.InternalError: Dst tensor is not initialized.
网络使用GPU训练时,一般当GPU显存被占满的时候会出现这个错误
解决的方法,就是在训练命令执行之前,首先执行下面的命令行:
Windows SET CUDA_VISIBLE_DEVICES=0
Linux export CUDA_VISIBLE_DEVICES=0
然后训练就会很顺利的执行下去