深度学习(cudnn加速)

cudnn为网络每一卷积层选最优实现方法,加速网络训练。

设置如下:

torch.backends.cudnn.benchmark = True

加速条件如下:

1. 输入数据在训练过程中一般不变化。

2. 数据量较大,并可以同时加载到GPU内存中。

3. 训练次数比较多。

posted @ 2023-10-09 23:36  Dsp Tian  阅读(54)  评论(0编辑  收藏  举报