Cudnn v5.1与V 6.0的特性

CUDNN v6引入了空洞卷积、一维FFT卷积、cudnnReduceTensor接口、deterministic max pooling模式等新特性,同时增强了对8位整型数据推理的支持。移除了部分旧版API,并修复了Dilated Convolution在Kepler GPU上的错误。CUDNN API遵循向后兼容策略,逐步淘汰旧版本接口。
摘要由CSDN通过智能技术生成

V5.1

新增特性

  1. 针对3*3以及5*5de 卷积核,添加了新的Winograd卷积算法,提供前向和后向计算。可以通过使用CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED,CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD_NONFUSED以及CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD_NONFUSED触发使用。
  2. cudnnConvolutionBackwardData以及cudnnConvolutionBackwardFilter扩展了针对FP16计算的支持
  3. 为很多常用的例子优化了LRN的性能

cudnn5.1与cudnn5.0接口保持完全兼容

V6

新增特性

  1. 空洞卷积(也称为膨胀卷积)Dilated Convolutions。目前在Cudnn中添加了对Dilated Convolution的支持,这一支持并未改变API。之前在Convolution Descriptor中未使用的域“upscale”,被定义成允许用户指定每个维度上的膨胀因子。当前对dilation的支持可以通过一下代码路径获取,
  2. Forward:CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM
  3. Backward Data: CUDNN_CONVOLUTION_BWD_DATA_ALOGO_0
  4. Backward Filter: CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0

  5. 新添加的接口cudnnConvolutionBiasActivationForward允许单个kernel中混合Convolution, bias以及activation操作。目前只支持单通道bias以及RELU激活函数。

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值