Mathematica中文论坛-非官方

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 1732|回复: 0

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

[复制链接]

534

主题

608

帖子

3055

积分

论坛元老

Rank: 8Rank: 8

积分
3055
发表于 2024-4-18 11:32:43 | 显示全部楼层 |阅读模式
这个问题,看了很多,
01.有的是说pytorch,cuda和驱动不匹配。
02.有的是说,显存不够。
03.有的是让不用cudnn进行训练。


01.重新安装
02.换卡,或减小训练的batchsize
03. torch.backends.cudnn.enabled = False

不一定好用。

我的问题。
我的数据,是在普通环境下生成的。(base)
我用数据进行训练, 是在conda的环境中实现的。(bfactor)

因为在base和bfactor中,两个np不一定版本一样。所以对应的,pytorch在训练时,觉得两个版本不一样(cudnn觉得是bfactor的npy版本,实际是base版本)
我的解决方案是在bfactor环境中,重新生成一次数据。
暂时看,是好用的。

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|Mathematica中文论坛-非官方 ( 辽ICP备16001491号-1

GMT+8, 2024-5-22 09:38 , Processed in 0.138528 second(s), 24 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表