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

cuda和cudnn安装完后, 运行程序显示报错
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
在把cudnn几个版本装了卸,卸了装之后,尝试把batch_size改小,从64改到24,训练正常运行了......
......为啥子提示不能完整点....

你可能感兴趣的:(RuntimeError: Unable to find a valid cuDNN algorithm to run convolution)