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