microsoftml.gpu_math: aceleración con NVidia CUDA

  • Artículo

Uso

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

Descripción

Implementación de CUDA de NVidia.

Argumentos

gpu_id

Identificador de dispositivo de GPU (configuración).

cu_dnn

Uso de cuDNN en GPU (configuración).

cu_dnn_algo

Opciones de optimización de cuDNN (configuración).

Vea también

avx_math, clr_math, mkl_math, sse_math