Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize #127

Open
fanruifeng opened this issue Aug 21, 2019 · 3 comments

Comments

@fanruifeng
Copy link

训练时 到 epoch 1的时候, 报错,
UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[node cnn/conv/conv1d/Conv2D (defined at E:\text-classification-cnn-rnn-master\cnn_model.py:50) = Conv2D[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="VALID", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](cnn/conv/conv1d/Conv2D-0-TransposeNHWCToNCHW-LayoutOptimizer, cnn/conv/conv1d/ExpandDims_1)]]
[[{{node optimize/Mean/_11}} = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_63_optimize/Mean", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
大家如果解决了 , 麻烦告诉下 谢谢 qq:1125729232

@xieqibin003
Copy link

我也碰到同样的问题了,我百度了说了cudnn的版本和tensorflow的版本不一致,但是我的cudnn的是7.4.2的,应该没问题吧?楼上的兄弟如果解决了麻烦留言一下哈

@StarZhai
Copy link

https://www.cnblogs.com/StarZhai/p/tensorflow-gpu.html
我是按照这个链接配置的GPU环境,运行这个项目没问题。你可以参考试试

@prozyworld
Copy link

win10下 用pycharm tensorflow-gpu 和 cuda 以及 cudnn版本对应上 什么都不用配置 版本对应可以参考官网https://www.tensorflow.org/install/source_windows 还有深度不够的话gpu版本不见得比cpu快 可以都尝试对比一下

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants