浅谈pytorch torch.backends.cudnn设置作用
作者:慢行厚积 时间:2022-06-20 16:25:06
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
来源:https://www.cnblogs.com/wanghui-garcia/p/11514502.html
标签:pytorch,torch.backends.cudnn
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Python实现压缩文件夹与解压缩zip文件的方法
2023-07-31 20:43:09
Python WXPY实现微信监控报警功能的代码
2021-05-15 02:07:20
![](https://img.aspxhome.com/file/2023/8/86178_0s.png)
Python关于__name__属性的含义和作用详解
2021-10-28 09:29:51
![](https://img.aspxhome.com/file/2023/2/77592_0s.png)
asp,php,.net使用301重定向方法
2007-09-26 14:05:00
python使用pymongo与MongoDB基本交互操作示例
2023-11-27 03:33:19
![](https://img.aspxhome.com/file/2023/5/88025_0s.png)
比较详细PHP生成静态页面教程
2023-10-14 18:54:31
38个Asp内置函数介绍
2008-11-27 16:25:00
perl批量查询ip归属地的方法代码
2023-08-11 22:53:54
PHP和JS之间的数据交互并处理
2023-05-25 00:57:08
页面制作中要注意的编码问题
2008-08-11 12:43:00
教你用Python实现简易版学生信息管理系统(含源码)
2022-07-21 23:40:00
Python实现Tab自动补全和历史命令管理的方法
2022-06-16 10:12:46
MySQL 语言参考
2007-11-21 20:37:00
js 操作cookie的教程
2008-03-05 13:08:00
一个免刷新页面的JavaScript日历
2007-12-26 12:57:00
![](https://img.aspxhome.com/file/UploadPic/200712/26/20071226131443435s.jpg)
三大原因:按照新HTML标准设计网页好处
2008-11-12 11:23:00
HTML+CSS 模仿Windows 7 桌面效果
2010-06-17 14:33:00
![](https://img.aspxhome.com/file/UploadPic/20106/17/1044050-15s.png)
多个python文件调用logging模块报错误
2021-03-30 23:30:24
python登录豆瓣并发帖的方法
2021-03-06 14:48:35
详解Django关于StreamingHttpResponse与FileResponse文件下载的最优方法
2021-04-13 08:17:01