microsoftml.gpu_math: Aceleración con NVidia CUDA

Usage

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

Description

Implementación de NVidia CUDA.

Argumentos

gpu_id

ID de dispositivo de la GPU (configuración).

cu_dnn

Usa cuDNN en la GPU (ajustes).

cu_dnn_algo

Opciones de optimización cuDNN (ajustes).