首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >无法运行tensorflow示例。CUBLAS误差

无法运行tensorflow示例。CUBLAS误差
EN

Stack Overflow用户
提问于 2017-11-20 20:48:47
回答 1查看 2.2K关注 0票数 0

大约一个月前,我就让这一切都开始运作了,现在我正试图重新运行所有的东西。我升级了所有的移动部件,在这里我尝试运行tensorflow示例:tensorflow git模型

设置:

  • Ubuntu 16.04
  • CUDA版本:版本8.0,V8.0.61 (能够运行deviceQuery等)
  • CUDNN_MAJOR 5(不管这是5.1还是6)
  • Tensorflow后端版本: 1.4.0 (无法运行示例)
  • keras版本: 2.1.1

这是我得到的错误:

2017年-11-20 15:18:00.209182: i tensorflow/core/common_runtime/gpu/gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (设备: 0,名称: GeForce GTX 1070,pci总线id: 0000:01:00.0,计算能力: 6.1) 2017-11-20 15:18:07.457924: e tensorflow/stream_executor/cuda/cuda_blas.cc:366]未能创建cublas句柄: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-20 15:18:07.474515: e tensorflow/stream_executor/cuda/cuda_blas.cc:366]未能创建cublas句柄: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-20 15:18:07.558178: e tensorflow/stream_executor/cudatensorflow/stream_executor/cuda/cuda_blas.cc:366] [ /cuda_blas.cc:366]创建cublas句柄失败: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-20 15:18:07.575864: e/cuda_blas.cc:366未能创建cublas句柄: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-20 15:18:07.593372: e tensorflow/stream_executor/cuda/cuda_blas.cc:366]未能创建cublas句柄: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-2015:18:07.610621: e tensorflow/stream_executor/cuda/cuda_blas.cc:366]未能创建cublas句柄: CUBLAS_STATUS_NOT_INITIALIZED 2017-11-20 15:18:08.431784: e tensorflow/stream_executor/cuda/cuda_dnn.cc:385]无法创建cudnn句柄: CUDNN_STATUS_INTERNAL_ERROR 2017-20 15:18:08.431816: e tensorflow/stream_executor/cuda/cuda_dnn.cc:352]无法破坏cudnn句柄: CUDNN_STATUS_BAD_PARAM 2017-11-20 15:18:08.431838: failed/core/核/conv_ops.cc:667]检查失败:CUDNN_STATUS_BAD_PARAM->parent()->GetConvolveAlgorithms(->GetConvolveAlgorithms)中止(核心抛出)

如有任何建议,将不胜感激。

编辑:这个问题:tensorflow运行错误与cublas是相似的,但答案对我没有用。

EN

回答 1

Stack Overflow用户

发布于 2017-11-22 22:56:12

这不是一个很好的解决方案,但它是唯一起作用的:

CUDNN 6和TF打得不好。所以如果你在任何地方都有图书馆,它就会崩溃。基本上,这是版本控制,而不是内存或路径。必须确保CUDNN v6在哪里找不到,TF是1.2版,而不是1.4版。

最后设置: Ubuntu 16.04。CUDA 8,CUDNN 5.1,TF 1.2。

安装tf版本可通过以下方式完成:

代码语言:javascript
复制
$ sudo apt install tensorflow-gpu==1.2 

这帮助我度过了黑暗时期:TensorFlow框架并列出了沿途的几个测试,以便进行健康检查。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/47400784

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档