RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

最新推荐文章于 2025-01-04 10:20:21 发布
最新推荐文章于 2025-01-04 10:20:21 发布
阅读量1.5k 收藏
点赞数 3
分类专栏： debug记录 文章标签： 深度学习 pytorch 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/onepunch_k/article/details/125091457
版权
博主在运行YOLOv5-6.1时遇到RuntimeError，提示无法找到有效的cuDNN算法进行卷积操作。经过排查，博主怀疑是CUDA版本不匹配，但确认无误后，发现是由于batch_size设置过大导致的问题。调整batch_size至合适大小后，问题得到解决。
摘要由CSDN通过智能技术生成

今天跑YOLOv5-6.1时出现这个问题：
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
网络上有些说是cuda版本不对之类的，但是我比较确定我这个没有问题，因为之前一直都可以弄。后来发现是batch_size弄太大了，调小以后就可以了

【深度学习】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
q742971636的博客
03-29 4939
正常安装https://blog.csdn.net/x1131230123/article/details/121798776后遇到这个问题咋办？ 原因可能是你的显卡显存容不下你的设置，把batch-size改小即可： python train.py --batch-size 2 --data kevin_firesmoke.yaml --img 1280 --epochs 500 --weight /home/kevin_xie/yolov5_version6/weights/yolov5l6.pt --
pytorch 训练 RuntimeError Unable to find a valid cuDNN algorithm to run convolution
mtl1994的博客
02-23 1万+
pytorch 训练 RuntimeError: Unable to find a valid cuDNN algorithm to run convolution pytorch 训练 RuntimeError: Unable to find a valid cuDNN algorithm to run convolution # 问题描述： python:3.95 pytorch:1.10.2 python train.py --img 640 --batch 64 --epochs 600 --d
【error】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
最新发布
qq_16721557的博客
01-04 235
故可以设置小一点的batch_size进行测试下，如果能够正确运行，那么就不需要进行更新cuda的版本。有很多博客让我们更新cuda，使显卡能够正确支持cuda的运行，这种方法。当出现错误是如下的问题时。
runtimeerror: unable to find a valid cudnn algorithm to run convolution
03-17
这个错误消息表示，程序在运行卷积运算时无法找到合适的 cuDNN 算法进行运算。 cuDNN 是一种 NVIDIA 公司提供的用于加速深度学习框架的库，它可以提高卷积运算的速度。如果程序无法找到合适的 cuDNN 算法，可能是...
RuntimeError:unable to find a valid cuDNN algorithm to convolution
m0_46630689的博客
07-26 1万+
由于虚拟环境都已经创建完成，目前是在Linux上进行安装，根据“链接2”中的安装代码，直接进行pytorch和cuda的安装。如果环境中已经装过其他版本的pytorch，应该能够自动卸载原版本，再装新的指定版本。经过验证，RTX3090安装上述版本的pytorch和cuda后，能够正常运行，不会报出上面的那个错误。2.其实问题更加简单，是模型的训练的batch-size训练过大了，调整更小，就可以了。解决办法：安装对应的cudnn，以及cuda，找到对应的torch框架，进行安装。
Unable to find a valid cuDNN algorithm to run convolution
出门吃三碗饭的博客
03-05 769
1GPU内存不够（但我之前遇到的GPU内存不够的报错，应该这样 ----- allocated 3.88/4.00 -----memory --不够 类似这样，和这个差挺多）总结：每个人遇到问题的报错不一样，但他们的解决方法是类似的，比如这里思维就是 去配置文件里面找一找，有没有与本机GPU情况不符的配置代码，改下就可以。后面自己结合上面的观点，推导了，大概就是项目配置文件里面对GPU的要求和我自己本机不太对应（我电脑就1个3050ti）使用上面的代码发现GPU是可以用的，可自己的torch版本对应。
【bug解决】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
热门推荐
千与编程
11-24 1万+
2.其实问题更加简单，是模型的训练的batch-size训练过大了，调整更小，就可以了。解决办法：安装对应的cudnn，以及cuda，找到对应的torch框架，进行安装。1.模型训练的环境中cudnn，CUDA的版本号不匹配。
pytorch bug: Unable to find a valid cuDNN algorithm to run convolution
weixin_54038731的博客
12-02 847
fcn复现报错
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值