cudnn加速计算

cudnn加速运算

　　torch.backends.cudnn.enabled = True
　　torch.backends.cudnn.benchmark = True

　　第一句话是说，使用的是非确定性算法，使用cudnn来寻找高效率的计算方式
　　第二句话是说，自动寻找最优的算法
　　一般情况下，当数据维度、类型变化不大的情况下这样使用有助于提高运算，
　　但是如果变化比较大，那么每次计算搜索最优算法的时间就会比较多，反而计算效率慢