microsoftml.gpu_math: Aceleração com NVidia CUDA

Usage

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

Description

Implementação NVidia CUDA.

Argumentos

gpu_id

ID do dispositivo da GPU (configurações).

cu_dnn

Use cuDNN na GPU (configurações).

cu_dnn_algo

opções de otimização cuDNN (configurações).