复现r2unet或rnn过程中报错RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

最新推荐文章于 2024-07-28 22:41:23 发布
最新推荐文章于 2024-07-28 22:41:23 发布
阅读量328 收藏
点赞数 6
分类专栏： python及pytorch异常错误处理 文章标签： rnn 人工智能 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/wwimhere/article/details/136344988
版权

今天尝试用pytorch复现了r2unet分割模型，这个模型在UNet中加入了recurrent block，也就是循环网络模块。

跑这个代码时报错了：

看上去像是cudnn版本问题，但是其实是循环网络模块需要的显存太大了。

我把batch size从20调到10之后，再次报错。

这次的报错就很明显了，说明了显存占用问题

我再次将batch size降到5.

代码这次跑起来

蓝海渔夫
关注
专栏目录
【深度学习】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
q742971636的博客
03-29 4679
正常安装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 --
RuntimeError: Placeholder storage has not been allocated on MPS device!解决方案
weixin_43178406的博客
01-12 2万+
本文主要介绍了RuntimeError: Placeholder storage has not been allocated on MPS device!解决方案，希望能对使用stable diffusion的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
nnUnet模型训练时报错：RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
weixin_44306345的博客
10-22 1072
跑模型时报错，遂解决记录。这个错误出现原因是模型所需显存过大，GPU显存不足导致的错误，这时需要将batch-size,patch-size改小即可：64-32等。
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
最新发布
weixin_44012667的博客
07-28 482
最后，如果您的GPU及其驱动程序不兼容cuDNN库，或者您不想使用cuDNN加速卷积计算，您可以禁用cuDNN。如果上述方法都无法解决问题，您可以尝试更改默认的卷积算法。这将启用Pytorch的自动卷积算法选择机制，它将根据运行时的硬件和数据情况自动选择最佳的卷积算法。Pytorch 无法找到有效的cuDNN算法来运行卷积|极客笔记 (deepinout.com)请注意，禁用cuDNN可能会降低卷积操作的性能。
在GPU上运行程序时报错：RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
qq_44955984的博客
02-04 5076
一、对于已经安装好了pytorch，cudn的情况下，一般都是因为自己笔记本的GPU内存不够导致的，解决方法如下： 1）调小batch_size的数值 2）换电脑吧，买GPU内存大的 3）最好还是用服务器跑深度学习的程序，在家的话可以在服务器和自己电脑上安装nomachine，这样就可以远程访问服务器了。 ...
代码报错：Unable to find a valid cuDNN algorithm to run convolution
iwill323的博客
11-25 2374
代码报错：Unable to find a valid cuDNN algorithm to run convolution 解决
【bug解决】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
千与编程
11-24 1万+
2.其实问题更加简单，是模型的训练的batch-size训练过大了，调整更小，就可以了。解决办法：安装对应的cudnn，以及cuda，找到对应的torch框架，进行安装。1.模型训练的环境中cudnn，CUDA的版本号不匹配。
RuntimeError: Error(s) in loading state_dict for ResNet: Unexpected key(s) in state_dict: “bn1.num
CSDN 精品推荐
01-09 902
在这个特定的情况下，错误信息中提到了多个“num_batches_tracked”键。这类键在某些 PyTorch 版本中存在，而在其他版本中可能不存在。方法会加载所有匹配的键，而忽略任何不匹配的键，从而避免了这个错误。但是，这种方法应谨慎使用，因为它可能掩盖模型结构和状态字典之间的不一致问题。这样做会使得 PyTorch 忽略不匹配的键，只加载那些与模型结构相匹配的键。这通常发生在模型的结构和要加载的状态字典不完全匹配时。解决这个问题的一个常见方法是在加载状态字典之前，明确地忽略不匹配的键。
Anaconda数据报错：Fatal error in launcher: Unable to create process using ‘“E:\Anaconda\python.exe“ “D:\
Deng872347348的博客
03-19 605
里面的pip.exe程序出现了问题，我这里直接删除，重新安装解决，上面已经手动删除pip.exe删除pip.exe，重新安装pip.exe用下面命令： 测试是否成功成功解决
OSError Unable to load weights from pytorch checkpoint file for pytorch_model.bin解决方案
热门推荐
weixin_43178406的博客
07-07 6万+
 本文主要介绍了OSError Unable to load weights from pytorch checkpoint file for pytorch_model-00003-of-00003.bin解决方案，希望能对学习大模型的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
RuntimeError reduce failed to synchronize cudaErrorAssert device-side assert triggered
cicylee的专栏
10-12 2237
RuntimeError: reduce failed to synchronize: cudaErrorAssert: device-side assert triggered 运行基于Unet的CrackForest代码时，出现该错误。 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-c2DA76xx-1602472447490)(C:\Users\dell\AppData\Roaming\Typora\typora-user-images\1602471396648.pn
Unet、AttentionUnet、R2Unet、R2AUet，可运行，包含数据集
03-24
构建Unet、AttentionUnet、R2Unet、R2AUet，做皮肤病的分割，使用的数据集是ISIC 2017。
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
pytorch bug: Unable to find a valid cuDNN algorithm to run convolution
weixin_54038731的博客
12-02 817
fcn复现报错
【报错】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
qq_58011370的博客
03-20 268
我的程序需要6G的GPU空间，一开始我使用的是显卡0，显卡0还剩余4G左右的空间，显然不够我的程序需要，所以就出现了上面的错误，在我换到显卡3（显卡3还剩余10G左右）以后，程序就可以正常执行了。这个错误的原因是咱们的。
RuntimeError: Error(s) in loading state_dict for UNet:
05-13
这个错误通常是由于加载模型时，模型的state_dict与当前代码中定义的模型结构不匹配导致的。可能的原因包括： 1. 模型定义发生了变化，导致state_dict与当前定义的模型结构不一致。 2. 加载的state_dict文件与当前模型结构不兼容，可能是由于使用了不同版本的PyTorch或者模型参数不同。 解决方法： 1. 确保模型定义没有发生变化，如果有变化需要重新训练模型或者手动调整state_dict的参数与模型结构一致。 2. 确保加载的state_dict文件与当前模型结构兼容，可以尝试使用相同版本的PyTorch或者手动修改state_dict参数。 以下是一个简单的示例，可以用来加载state_dict并检查其与当前模型结构是否匹配： ``` model = UNet() # 创建模型 state_dict = torch.load(PATH) # 加载 state_dict model.load_state_dict(state_dict) # 加载 state_dict # 检查模型参数是否匹配 for name, param in model.named_parameters(): if name in state_dict: if state_dict[name].shape != param.shape: print(f"Error: Parameter {name} does not match!") else: print(f"Warning: Parameter {name} not found in state_dict!") ```
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝海渔夫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值