cuda和cudnn安装完后, 运行程序显示报错
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
在把cudnn几个版本装了卸,卸了装之后,尝试把batch_size改小,从64改到24,训练正常运行了......
本文转载自:http://events.jianshu.io/p/75f901f94fee
博客作者在安装CUDA和cuDNN后遇到运行程序时的RuntimeError,提示找不到有效的cuDNN算法进行卷积运算。尝试了不同版本的cuDNN并调整batch_size后,发现将batch_size从64减至24使得训练可以正常进行。这个问题可能与cuDNN版本不兼容或者资源限制有关。
cuda和cudnn安装完后, 运行程序显示报错
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
在把cudnn几个版本装了卸,卸了装之后,尝试把batch_size改小,从64改到24,训练正常运行了......
本文转载自:http://events.jianshu.io/p/75f901f94fee
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
5128
1万+
2773

被折叠的 条评论
为什么被折叠?