浅谈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选择排序算法实例总结
Jul 01 Python
Python应用03 使用PyQT制作视频播放器实例
Dec 07 Python
python通过ffmgep从视频中抽帧的方法
Dec 05 Python
在python中使用with打开多个文件的方法
Jan 07 Python
Pycharm设置utf-8自动显示方法
Jan 17 Python
python实现比较类的两个instance(对象)是否相等的方法分析
Jun 26 Python
Python 字符串、列表、元组的截取与切片操作示例
Sep 17 Python
pygame实现五子棋游戏
Oct 29 Python
python 中值滤波,椒盐去噪,图片增强实例
Dec 18 Python
python能否java成为主流语言吗
Jun 22 Python
Flask处理Web表单的实现方法
Jan 31 Python
python 30行代码实现蚂蚁森林自动偷能量
Feb 08 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
实现 win2003 下 mysql 数据库每天自动备份
2006/12/06 PHP
php实现在线考试系统【附源码】
2018/09/18 PHP
PHP生成指定范围内的N个不重复的随机数
2019/03/18 PHP
Discuz不使用插件实现简单的打赏功能
2019/03/21 PHP
laravel框架模板之公共模板、继承、包含实现方法分析
2019/08/30 PHP
linux mint下安装phpstorm2020包括JDK部分的教程详解
2020/09/17 PHP
js用图作提交按钮或超连接
2008/03/26 Javascript
javascript 函数调用的对象和方法
2010/07/01 Javascript
JavaScript获取表单内所有元素值的方法
2015/04/02 Javascript
初步认识JavaScript函数库jQuery
2015/06/18 Javascript
基于jQuery仿淘宝产品图片放大镜代码分享
2020/06/23 Javascript
用jQuery获取table中行id和td值的实现代码
2016/05/19 Javascript
全面解析JavaScript中的valueOf与toString方法(推荐)
2016/06/14 Javascript
vue组件父子间通信详解(三)
2017/11/07 Javascript
Nodejs实现爬虫抓取数据实例解析
2018/07/05 NodeJs
vue仿element实现分页器效果
2018/09/13 Javascript
javascript原型链学习记录之继承实现方式分析
2019/05/01 Javascript
微信小程序实现录制、试听、上传音频功能(带波形图)
2020/02/27 Javascript
解决iView Table组件宽度只变大不变小的问题
2020/11/13 Javascript
Python标准库与第三方库详解
2014/07/22 Python
Python中index()和seek()的用法(详解)
2017/04/27 Python
python smtplib发送带附件邮件小程序
2018/05/22 Python
python使用__slots__让你的代码更加节省内存
2018/09/05 Python
Python 脚本实现淘宝准点秒杀功能
2019/11/13 Python
Python脚本导出为exe程序的方法
2020/03/25 Python
美国办公用品购物网站:Quill.com
2016/09/01 全球购物
AHAVA美国官方网站:死海海泥护肤品牌
2016/10/18 全球购物
奥地利网上现代灯具和灯饰店:Lampenwelt.at
2018/01/29 全球购物
华为C++笔试题
2014/08/05 面试题
总裁秘书岗位职责
2013/12/04 职场文书
自我评价格式
2014/01/06 职场文书
工艺工程师岗位职责
2014/03/04 职场文书
工作检讨书500字
2014/10/19 职场文书
MySQL创建定时任务
2022/01/22 MySQL
「约定的梦幻岛」作画发布诺曼生日新绘
2022/03/21 日漫
java中如何截取字符串最后一位
2022/07/07 Java/Android