浅谈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 相关文章推荐
浅谈pycharm下找不到sqlalchemy的问题
Dec 03 Python
利用Python查看微信共同好友功能的实现代码
Apr 24 Python
Pandas之MultiIndex对象的示例详解
Jun 25 Python
python之mock模块基本使用方法详解
Jun 27 Python
解决Django后台ManyToManyField显示成Object的问题
Aug 09 Python
Django 对IP访问频率进行限制的例子
Aug 30 Python
计算pytorch标准化(Normalize)所需要数据集的均值和方差实例
Jan 15 Python
Python sep参数使用方法详解
Feb 12 Python
Django查询优化及ajax编码格式原理解析
Mar 25 Python
Python使用matplotlib绘制圆形代码实例
May 27 Python
Python爬虫入门案例之回车桌面壁纸网美女图片采集
Oct 16 Python
使用python求解迷宫问题的三种实现方法
Mar 17 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/01/22 PHP
PHP时间戳 strtotime()使用方法和技巧
2013/10/29 PHP
PHP实现数组递归转义的方法
2014/08/28 PHP
Thinkphp搜索时首页分页和搜索页保持条件分页的方法
2014/12/05 PHP
PHP+ajax实现二级联动菜单功能示例
2018/08/10 PHP
页面使用密码保护代码
2013/04/10 Javascript
jQuery简单实现图片预加载
2015/04/20 Javascript
jquery实现浮动在网页右下角的彩票开奖公告窗口代码
2015/09/04 Javascript
JavaScript事件学习小结(三)js事件对象
2016/06/09 Javascript
ECMAScript6快速入手攻略
2016/07/18 Javascript
浅谈javascript中的Function和Arguments
2016/08/30 Javascript
js获取浏览器高度 窗口高度 元素尺寸 偏移属性的方法
2016/11/21 Javascript
npm国内镜像 安装失败的几种解决方案
2017/06/04 Javascript
用最简单的方法判断JavaScript中this的指向(推荐)
2017/09/04 Javascript
VUE2 前端实现 静态二级省市联动选择select的示例
2018/02/09 Javascript
Vue二次封装axios为插件使用详解
2018/05/21 Javascript
使用vue根据状态添加列表数据和删除列表数据的实例
2018/09/29 Javascript
react实现antd线上主题动态切换功能
2019/08/12 Javascript
vue flex 布局实现div均分自动换行的示例代码
2020/08/05 Javascript
SublimeText 2编译python出错的解决方法(The system cannot find the file specified)
2013/11/27 Python
Python2.x版本中基本的中文编码问题解决
2015/10/12 Python
Django开发中的日志输出的方法
2018/07/02 Python
python pandas移动窗口函数rolling的用法
2020/02/29 Python
python读取当前目录下的CSV文件数据
2020/03/11 Python
使用Keras构造简单的CNN网络实例
2020/06/29 Python
Python gevent协程切换实现详解
2020/09/14 Python
5分钟快速掌握Python定时任务框架的实现
2021/01/26 Python
python可视化分析的实现(matplotlib、seaborn、ggplot2)
2021/02/03 Python
跑鞋、网球鞋、网球拍、服装及装备:Holabird Sports
2016/09/19 全球购物
欧姆龙医疗保健与医疗产品:Omron Healthcare
2020/02/10 全球购物
美国沃尔玛网上超市:Walmart
2020/08/14 全球购物
.NET笔试题(20个问题)
2016/02/02 面试题
思想政治自我鉴定
2013/10/06 职场文书
出国留学自荐信
2013/10/25 职场文书
求职信写作要突出重点
2014/01/01 职场文书
爸爸的三轮车观后感
2015/06/16 职场文书