浅谈pytorch torch.backends.cudnn设置作用


Posted in Python onFebruary 20, 2020

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 相关文章推荐
python在命令行下使用google翻译(带语音)
Jan 16 Python
pyqt和pyside开发图形化界面
Jan 22 Python
Python基于动态规划算法计算单词距离
Jul 25 Python
python3.x实现发送邮件功能
May 22 Python
python自动发送邮件脚本
Jun 20 Python
python 判断矩阵中每行非零个数的方法
Jan 26 Python
Django logging配置及使用详解
Jul 23 Python
python读取指定字节长度的文本方法
Aug 27 Python
Kears+Opencv实现简单人脸识别
Aug 28 Python
通过Python编写一个简单登录功能过程解析
Sep 04 Python
Python openpyxl模块原理及用法解析
Jan 19 Python
Pyecharts 中Geo函数常用参数的用法说明
Feb 01 Python
Python sqlite3查询操作过程解析
Feb 20 #Python
python利用datetime模块计算程序运行时间问题
Feb 20 #Python
pytorch数据预处理错误的解决
Feb 20 #Python
Python异常继承关系和自定义异常实现代码实例
Feb 20 #Python
Python安装与卸载流程详细步骤(图解)
Feb 20 #Python
PyCharm 专业版安装图文教程
Feb 20 #Python
Python3.6安装卸载、执行命令、执行py文件的方法详解
Feb 20 #Python
You might like
PHP对字符串的递增运算分析
2010/08/08 PHP
基于PHP的cURL快速入门教程 (小偷采集程序)
2011/06/02 PHP
PHP 7的一些引人注目的新特性简单介绍
2015/11/08 PHP
PHP函数shuffle()取数组若干个随机元素的方法分析
2016/04/02 PHP
AES加解密在php接口请求过程中的应用示例
2016/10/26 PHP
利用PHP访问MySql数据库的逻辑操作以及增删改查的实例讲解
2017/08/30 PHP
JSQL 基于客户端的成绩统计实现方法
2010/05/05 Javascript
Extjs显示从数据库取出时间转换JSON后的出现问题
2012/11/20 Javascript
JQuery判断子iframe何时加载完成解决方案
2013/08/20 Javascript
有效提高JavaScript执行效率的几点知识
2015/01/31 Javascript
jquery实现适用于门户站的导航下拉菜单效果代码
2015/08/24 Javascript
Google 地图控件集详解及实例代码
2016/08/06 Javascript
微信小程序之选项卡的实现方法
2017/09/29 Javascript
基于input动态模糊查询的实现方法
2017/12/12 Javascript
微信小程序实现页面跳转传值以及获取值的方法分析
2017/12/18 Javascript
JS高级技巧(简洁版)
2018/07/29 Javascript
JS判断用户用的哪个浏览器实例详解
2018/10/09 Javascript
jqGrid表格底部汇总、合计行footerrow处理
2019/08/21 Javascript
Vue Cli3 打包配置并自动忽略console.log语句的方法
2020/04/23 Javascript
小程序富文本提取图片可放大缩小
2020/05/26 Javascript
vue实现单一筛选、删除筛选条件
2020/10/26 Javascript
python实现class对象转换成json/字典的方法
2016/03/11 Python
深入理解python中的浅拷贝和深拷贝
2016/05/30 Python
python中print的不换行即时输出的快速解决方法
2016/07/20 Python
Windows下Python2与Python3两个版本共存的方法详解
2017/02/12 Python
Python wxPython库使用wx.ListBox创建列表框示例
2018/09/03 Python
详解python实现小波变换的一个简单例子
2019/07/18 Python
Pycharm 2020年最新激活码(亲测有效)
2020/09/18 Python
python 实现人和电脑猜拳的示例代码
2020/03/02 Python
澳大利亚领先的时尚内衣零售商:Bras N Things
2020/07/28 全球购物
如何清空Session
2015/02/23 面试题
文化宣传方案
2014/03/13 职场文书
毕业论文答辩开场白和答辩技巧
2015/05/27 职场文书
幼儿园中班班级总结
2015/08/10 职场文书
python实现简单区块链结构
2021/04/25 Python
详解Python魔法方法之描述符类
2021/05/26 Python