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

最新推荐文章于 2023-12-22 09:05:17 发布

galaxyzwj

最新推荐文章于 2023-12-22 09:05:17 发布

阅读量678

点赞数

分类专栏： # tensorflow 文章标签：训练模型报错

本文链接：https://blog.csdn.net/zz531987464/article/details/103750061

版权

tensorflow 专栏收录该内容

22 篇文章 1 订阅

订阅专栏

训练模型时老是报上面的错误，不断修改batch size也没有效果

执行命令 watch -n 0.1 nvidia-smi 可以看到

看到GPU 显存爆了。。。

这里给出网上找到的一个适用于tensorflow-gpu2.0.0的方法，在前面加上如下代码即可：

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

galaxyzwj

关注关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try…

01-06

使用inception-V3批处理文件retrain.bat进行预训练时，发现报错“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try…”于是打开retrain.py文件，在前面加上如下代码： from tensorflow.compat.v1 import ConfigProto from tensorflow.compat.v1 import InteractiveSession config = ConfigProto() config.gpu_o

解决Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

qq_38835585的博客

09-23

8338

问题如下： 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 conv2d_1/convolution (defined at C:\Us

参与评论您还未登录，请先登录后发表或查看评论

Failed to get convolution algorithm. This is probably because cuDNN failed..(TensorFlow和keras显存不足报错)

nohopenolove的博客

11-25

2066

在跑深度学习程序时，用到TensorFlow或者keras时候，经常会报一个错误： tensorflow.python.framework.errors_impl. UnknownError: 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.

“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误解决方案

得峰的专栏 [网络收藏]

08-18

747

最近在使用TF2.0。运行程序出现以下错误。 Failed to get convolution algorithm. This is probably because cuDNN failed to initialize 一开始怀疑是CUDA和CuDNN配置错误（要求版本匹配）。反复试验后，还是有这个错误。最后发现可能是GPU显存不足造成的。需要在程序前加以下一段代码： from tensorflow.compat.v1 import ConfigProto from tensorflow.

tensorflow卷积报错Failed to get convolution algorithm. This is probably because cuDNN failed t

集电极

07-05

4699

tensorflow2.0卷积报错Failed to get convolution algorithm. This is probably because cuDNN failed t 说明：环境是tensorflow2.0.0,测试GPU没有问题。但在运行模型出现错误。 UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try lookin

Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so

12-22

This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. 主要原因是版本问题，还有可能是没有安装cuda / cudnn ,具体可在pycharm的设置中查看

CUDNN_Library.rar_cudnn_cudnn 库文件_yeshet

09-23

1. **卷积操作优化**：CUDNN 提供了多种卷积算法，包括直接卷积、Fused Convolution（融合卷积）、Winograd 算法等，这些算法针对不同的硬件配置和数据规模进行了优化，可以充分利用 GPU 的并行计算能力。...

深度学习支持包 Cudnn11.x

05-16

深度学习支持包Cudnn11.x是针对深度学习领域的一款关键加速库，它由NVIDIA公司开发，专为利用CUDA编程环境的GPU提供优化的深度神经网络（DNN）运算。CUDNN，全称为CUDA Deep Neural Network library，是深度学习框架...

cudnn-11.2-windows-x64-v8.1.0.77.zip

07-01

**CUDNN (CUDA Deep Neural Network) 简介** CUDNN，全称为CUDA深层神经网络库（CUDA Deep Neural Network Library），是由NVIDIA公司开发的一款高性能、优化的深度学习库，专门用于加速基于GPU（图形处理器单元）...

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决办法

牧野渔樵

11-08

767

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决办法在网上看了很多说法，总结就两种情况： cudnn和tensorflow的版本不相符 GPU内存不足造成第一种情况：需要查出cuda、cudnn、tensorflow的版本对应关系，cuda版本只能是9.0,...

“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法