首页 > 解决方案 > keras 与 TensorFlow GPU、CUDA_ERROR_LAUNCH_FAILED 超参数搜索

问题描述

我正在使用带有 TensorFlow 后端的 Keras。

我编写搜索脚本来调整我的 CuDDNNLSTM 超参数。

创建约 10 个不同的 CuDDNNLSTM 网络后,我收到错误: tensorflow\stream_executer\cuda\cuda_driver.cc:1108 could not synchronize on CUDA context: CUDA_ERROR_LAUNCH_FAILED在搜索过程中。在 : tensorflow\python\client\session.py 在 _run,_do_run,_do_call

操作系统:WIN10 64x

蟒蛇:3.6.5

Keras 版本:2.1.6

张量流/GPU:1.10.0

CUDA:9.0

cudn:7.3

显卡:GeForce GTX 1080 Ti

有人会遇到那个问题吗?

标签: pythontensorflowkerasgpu

解决方案


推荐阅读