cudnn.benchmark

tang-02038 个月前
pytorch·cudnn.benchmark
torch.backends.cudnn.benchmark 作用相关参数如果输出结果为True，则表示cuDNN已成功设置为PyTorch的后端。这意味着PyTorch将利用cuDNN加速深度神经网络的计算。