Compartilhar via

microsoftml.gpu_math: aceleração com a NVidia CUDA

  • Artigo

Uso

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

Descrição

Implementação da NVidia CUDA.

Argumentos

gpu_id

Identificação do dispositivo de GPU (configurações).

cu_dnn

Use cuDNN na GPU (configurações).

cu_dnn_algo

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

Confira também

avx_math, clr_math, mkl_math, sse_math