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