MindSpore中是否有类似torch.backends.cudnn.benchmark函数功能
时间: 2024-05-24 17:13:05 浏览: 109
浅谈pytorch torch.backends.cudnn设置作用
5星 · 资源好评率100%
MindSpore中有类似于torch.backends.cudnn.benchmark的功能,即使用mindspore.context.set_context(mode=mindspore.context.PYNATIVE_MODE, device_target="GPU", enable_graph_kernel=True)可以启用MindSpore的图形核心编译器,从而提高GPU的性能。
阅读全文