Pytorch Bug: RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

已于 2022-06-20 18:27:06 修改
阅读量809 收藏
点赞数
分类专栏： 图像处理 医学图像分割 pytorch 文章标签： pytorch 深度学习 神经网络 人工智能 机器学习
于 2022-06-20 18:26:00 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_46695905/article/details/125377348
版权

Pytorch Bug: RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

最近在做一个多器官分割的任务，总共22个类别器官！ 使用 torch.nn.CrossEntropyLoss() 和 自定义的 DiceLoss作为损失函数，多机多卡运行。在运行的过程中产生了这个错误，记录一下。
原因产生在 torch.nn.CrossEntropyLoss() 损失函数。

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

在这里插入图片描述

CUDA error: an illegal memory access was encountered

在这里插入图片描述

看很多人都说是 torch 版本问题。
首先检查了一下版本，确定没有问题。
之后Debug了几个小时，终于解决！！！
主要原因是 Label 的类别数与模型输出不一致！
比如 Label 包含 22 个器官，加上背景，其应该是 23 个类别，因此模型输出通道应该是 23，而不是 22 ！！！

菜鸟成长乐园
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
解决Pytorch的cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
Jurio的博客
04-28 3281
解决Pytorch的cuDNN error: CUDNN_STATUS_NOT_INITIALIZED，总结四种可能原因及解决方法和链接
yolov5训练集出现报错：RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERRORYou can try to repro this excep
qq_53545309的博客
04-12 742
yolovs中train.py报错
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
qq_43509393的博客
05-07 5864
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED版本问题算力不足pip问题数据位置其他方案 pytorch训练模型的时候出现的报错信息，尝试了CSDN中的很多办法，在pytorch forums里也试了ptrblck给出的方案，stack overflow里也有人给出了回答， 在这里总结一下。 版本问题 检查自己的torch、torchvision、cuda、python版本是否匹配。贴一个链接。 算力不足 环境没有问题，可能是算力不足， wat
【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
Hawley的博客
07-10 1万+
【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 文章目录【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED问题描述解决思路1、验证cuda是否可用2、如果cuda不可用，则查看cuda版本3、如果cuda可用，则查看pytorch版本总结 问题描述 在git上下载了相关的代码准备在服务器上跑一下，然后发现在之前配好的conda虚拟环境下跑不了，版本太高了，所以就重新配了一个虚拟环境，结果遇到了以
已解决！！！RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
qq_41973051的博客
08-08 1388
已经提示了loss不能反传“loss.backward()”，以及“Assertion `t >= 0 && t < n_classes` failed”，需要检查分类类别数量是否与实际数据的类别数量一致，那怎么检查呢？
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案
热门推荐
weixin_43178406的博客
06-12 8万+
本文主要介绍了RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案，希望能对学习pytorch的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
解决 RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
最新发布
m0_51870122的博客
04-14 1847
在学习和调试Moco代码时运行到查阅了很多资料可能原因如下，供大家自查。
报错：RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
秦耘
09-12 1万+
1、惨痛教训：一般显存溢出报out of memory之类，修改了代码中batch-size大小（忘记自己已经配置过默认参数）未解决。所以便认为是cuda配置问题，多方检查确认cuda、cudnn配置无误，也经历过更换cudnn，都无果。最终想起配置过的默认参数未改，改后正常。所以在查那么久，原因五花八门，都解决不了问题，此报错极可能根本与版本、配置无关，仅仅算力不足而已。 ![在这里插入图片描述](https://img-blog.csdnimg.cn/20200912160034881.png?x-os
发生异常: RuntimeError cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
09-08
发生异常: RuntimeError cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 是由于cuDNN库没有正确初始化导致的。cuDNN是一个用于深度学习的GPU加速库，它提供了许多优化的算法和函数来加速神经网络的训练和推理过程。...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值