通过

microsoftml.gpu_math：通过 NVidia CUDA 加速

  • 项目

使用情况

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

说明

NVidia CUDA 实现。

参数

gpu_id

GPU 设备 ID（设置）。

cu_dnn

在 GPU 上使用 cuDNN（设置）。

cu_dnn_algo

cuDNN 优化选项（设置）。

请参阅

avx_math, clr_math, mkl_math, sse_math