浅谈pytorch torch.backends.cudnn设置作用
更新时间：2020年02月20日 10:49:55 作者：慢行厚积
今天小编就为大家分享一篇浅谈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
以上这篇浅谈pytorch torch.backends.cudnn设置作用就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在本篇内容里小编给大家分享的是关于python print出共轭复数的方法总结内容，有需要的读者们可以学习下。2019-06-06
这篇文章主要介绍了Python 集合 pop()函数的使用方法，文中有详细的代码实例，讲解的非常清楚，具有一定的参考价值,需要的朋友可以参考下2023-07-07
本文主要介绍了Python脚本修改Maya ASCII文件路径方法实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
今天小编就为大家分享一篇Django项目中包含多个应用时对url的配置方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要介绍了浅谈一下Python中的with用法,有一些任务，可能事先需要设置，事后做清理工作。对于这种场景，Python的with语句提供了一种非常方便的处理方式,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了Python跑循环时内存泄露的解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-01-01
在50行代码内来实现一个贪吃蛇，具体实现代码如下，感兴趣的朋友可以参考下哈，希望可以帮助到你2013-04-04
这篇文章主要介绍了Python实现二分法算法实例,本文先是介绍了算法规则,然后给出了实现例子,需要的朋友可以参考下2015-02-02
Pygame 中提供了一个draw模块用来绘制一些简单的图形状，比如矩形、多边形、圆形、直线、弧线等，本文就详细的介绍一下如何使用2021-11-11
图像分割是将图像分成若干具有独特性质的区域并提取感兴趣目标的技术和过程，这篇文章将详细讲解漫水填充分割应用，感兴趣的小伙伴可以了解一下2023-01-01
最新评论