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の実装。

引数

gpu_id

GPUデバイスID(設定)。

cu_dnn

GPUの設定でcuDNNを使いましょう。

cu_dnn_algo

cuDNNの最適化オプション(設定)。