共用方式為


microsoftml.gpu_math:使用 NVidia CUDA 加速

使用方式

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

Description

NVidia CUDA 實作。

引數

gpu_id

GPU 裝置識別碼 (設定)。

cu_dnn

在 GPU 上使用 cuDNN (設定)。

cu_dnn_algo

cuDNN 最佳化選項 (設定)。

另請參閱

avx_math, clr_math, mkl_math, sse_math