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 最適化オプション (設定)。