浅谈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 os模块介绍
Nov 30 Python
python利用标准库如何获取本地IP示例详解
Nov 01 Python
Python爬虫实例_利用百度地图API批量获取城市所有的POI点
Jan 10 Python
使用Python通过win32 COM打开Excel并添加Sheet的方法
May 02 Python
在linux下实现 python 监控usb设备信号
Jul 03 Python
django获取from表单multiple-select的value和id的方法
Jul 19 Python
pytorch 使用单个GPU与多个GPU进行训练与测试的方法
Aug 19 Python
Python threading的使用方法解析
Aug 28 Python
Python 用三行代码提取PDF表格数据
Oct 13 Python
Django 解决model 反向引用中的related_name问题
May 19 Python
浅谈Tensorflow加载Vgg预训练模型的几个注意事项
May 26 Python
在Django中使用MQTT的方法
May 10 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 ob缓存以及ob函数原理实例解析
2020/11/13 PHP
Avengerls vs KG BO3 第三场2.18
2021/03/10 DOTA
推荐10个超棒的jQuery工具提示插件
2011/10/11 Javascript
java与javascript之间json格式数据互转介绍
2013/10/29 Javascript
JS+CSS模拟可以无刷新显示内容的留言板实例
2015/03/03 Javascript
javascript解析xml实现省市县三级联动的方法
2015/07/25 Javascript
jquery实现手机号码选号的方法
2015/07/31 Javascript
学习JavaScript设计模式之模板方法模式
2016/01/20 Javascript
利用Node.js编写跨平台的spawn语句详解
2017/02/12 Javascript
利用C/C++编写node.js原生模块的方法教程
2017/07/07 Javascript
JS匿名函数内部this指向问题详析
2019/05/10 Javascript
小程序如何自主实现拦截器的示例代码
2019/11/04 Javascript
详解Vue串联过滤器的使用场景
2020/04/30 Javascript
Python中比较特别的除法运算和幂运算介绍
2015/04/05 Python
分析Python的Django框架的运行方式及处理流程
2015/04/08 Python
浅谈Python中的闭包
2015/07/08 Python
Python入门之后再看点什么好?
2018/03/05 Python
对Tensorflow中的矩阵运算函数详解
2018/07/27 Python
使用Python完成15位18位身份证的互转功能
2019/11/06 Python
Python实现计算图像RGB均值方式
2020/06/04 Python
浅析Python 简单工厂模式和工厂方法模式的优缺点
2020/07/13 Python
python如何快速拼接字符串
2020/10/28 Python
Python 利用argparse模块实现脚本命令行参数解析
2020/12/28 Python
html5中如何将图片的绝对路径转换成文件对象
2018/01/11 HTML / CSS
美国著名手表网站:Timepiece
2017/11/15 全球购物
Jones New York官网:美国女装品牌,受白领女性欢迎
2019/11/26 全球购物
企划主管岗位职责
2013/12/12 职场文书
十佳美德少年事迹材料
2014/02/05 职场文书
致铅球运动员加油稿
2014/02/13 职场文书
毕业生就业推荐表自我鉴定
2014/03/20 职场文书
运动员口号
2014/06/09 职场文书
省委召开党的群众路线教育实践活动总结大会报告
2014/10/21 职场文书
承兑汇票转让证明怎么写?
2014/11/30 职场文书
大足石刻导游词
2015/02/02 职场文书
公司年会主持词范文!
2019/05/07 职场文书
win11电脑关机鼠标灯还亮怎么解决? win11关机后鼠标灯还亮解决方法
2023/01/09 数码科技