microsoftml.gpu_math:用NVidia CUDA加速

Usage

microsoftml.gpu_math(gpu_id: numbers.Real = -1,
    cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')

Description

NVidia CUDA 实现。

Arguments

gpu_id

显卡设备ID(设置)。

cu_dnn

在GPU上使用cuDNN(设置)。

cu_dnn_algo

cuDNN 优化选项(设置)。