Check failed: stream->parent()->GetConvolveAlgorithms 解决方法(适用于CUDNN版本6的)

来源:互联网 发布:网络直播监督平台注册 编辑:程序博客网 时间:2024/05/18 02:51

在验证HED网络时,提示
Check failed: stream->parent()->GetConvolveAlgorithms
等信息,但是重新运行其他程序可用。

检查CUDNN版本为6.0,想起来之前某个帖子说到可能是
tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
的锅,将该值设置为0.7后正常运行。

阅读全文
0 0
原创粉丝点击