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

文章讲述了在运行时遇到RuntimeError:UnabletofindavalidcuDNNalgorithmtorunconvolution的问题。解决方案是设置torch.backends.cudnn.benchmark=True,虽然可以提高训练速度,但可能导致硬件间的性能差异,且会增加GPU的内存使用。
摘要由CSDN通过智能技术生成

1. bug内容
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

2. 解决办法
设置
torch.backends.cudnn.benchmark = True
加上这一行后bug解决了,而且训练还更快了(但是在不同硬件下可能情况不同,因为我在本地电脑机上这样设置后训练更慢了,但是在服务器上更快了)

其他
而且这样设置后悔导致GPU的利用率和内存使用率都增加。影响中设置前GPU Memory-Usage大概在28000M左右

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值