浅谈pytorch torch.backends.cudnn设置作用
cuDNN使用非确定性算法，并且可以使用torch.backends.cudnn.enabled=False来进行禁用
如果设置为torch.backends.cudnn.enabled=True，说明设置为使用使用非确定性算法
然后再设置：
torch.backends.cudnn.benchmark=true
那么cuDNN使用的非确定性算法就会自动寻找最适合当前配置的高效算法，来达到优化运行效率的问题
一般来讲，应该遵循以下准则：
如果网络的输入数据维度或类型上变化不大，设置 torch.backends.cudnn.benchmark=true 可以增加运行效率；
如果网络的输入数据在每次iteration都变化的话，会导致cnDNN每次都会去寻找一遍最优配置，这样反而会降低运行效率。
所以我们经常看见在代码开始出两者同时设置：
torch.backends.cudnn.enabled=True
torch.backends.cudnn.benchmark=True
以上这篇浅谈pytorchtorch.backends.cudnn设置作用就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。