|
问题描述:
CuDNN: could not create cudnn hadle: CUDNN_STATUS_NOT_INITIALIZED
Check failed: stream->parent()->GetConvolveAlgoritms( conv_parameters.ShouldIncludeNinogradeNondusedAlgo<T>(), &algoriyms)
已放弃(核心已转储)
环境介绍:
系统:Ubuntu 16.04
CUDA: CUDA-9.0
CuDNN: CuDNN-7.1 for CUDA9.1 (其实写到这里就知道肯定会出问题的)
我在安装tensorflow1.7的时候,使用的CUDA-9.1和CuDNN-7.1 for CUDA9.1,但是实际在import tensorflow的时候,提示的是找不到 libcudart.so.9.0,所以随后将CUDA-9.1卸载,再安装CUDA-9.0,在还没有更改CuDNN的版本下,成功的 import tensorflow。
在跑程序的时候,因为程序中使用到了CNN,就得到了以上的错误,看网上的说法是如果模型没有使用CNN,应该不会报错,但是我没有去尝试了。错误提示的是:CUDNN没有初始化,再下面一句话是“可能驱动版本不足”。我将CUDNN版本换成CUDNN for CUDA-9.0 之后边没有这个错误提示了。所以不是驱动的问题。
|
|