在验证HED网络时,提示
Check failed: stream->parent()->GetConvolveAlgorithms
等信息,但是重新运行其他程序可用。
检查CUDNN版本为6.0,想起来之前某个帖子说到可能是
tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
的锅,将该值设置为0.7后正常运行。
有趣的是,同样的代码,在M4000的机器上没有这个问题,在1060(6G)上却出现——估计是显存不足的锅。