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

顯示卡裝置ID(設定)。

cu_dnn

在 GPU 上使用 cuDNN(設定)。

cu_dnn_algo

cuDNN 優化選項(設定)。