小编Gno*_*ske的帖子

使用CUDNN_STATUS_ALLOC_FAILED的Tensorflow崩溃

一直在网上搜索数小时,没有任何结果,所以我想在这里问。

我正在尝试按照Sentdex的教程制作自动驾驶汽车,但是在运行模型时,会遇到很多致命错误。我已经在整个互联网上搜索了解决方案,许多似乎都遇到了同样的问题。但是,我发现的所有解决方案(包括此Stack-post)都不适合我。

这是我的软件:

  • Tensorflow:1.5,GPU版本
  • CUDA:9.0,带有补丁
  • CUDnn:7
  • Windows 10专业版
  • Python 3.6

硬件:

  • Nvidia 1070ti,带有最新驱动程序
  • 英特尔i5 7600K

这是崩溃日志:

2018-02-04 16:29:33.606903: E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_blas.cc:444] failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 2018-02-04 16:29:33.608872: E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_blas.cc:444] failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 2018-02-04 16:29:33.609308: E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_blas.cc:444] failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 2018-02-04 16:29:35.145249: E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_dnn.cc:385] could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED 2018-02-04 16:29:35.145563: E C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\stream_executor\cuda\cuda_dnn.cc:352] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM 2018-02-04 16:29:35.149896: F C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\kernels\conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)

这是我的代码:

 import tensorflow …
Run Code Online (Sandbox Code Playgroud)

python neural-network python-3.x tensorflow

1
推荐指数
3
解决办法
6548
查看次数

标签 统计

neural-network ×1

python ×1

python-3.x ×1

tensorflow ×1