Partage via

microsoftml.gpu_math : accélération avec NVidia CUDA

  • Article

Usage

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

Description

Implémentation NVidia CUDA.

Arguments

gpu_id

ID du périphérique GPU (paramètres).

cu_dnn

Utilisez cuDNN sur le GPU (paramètres).

cu_dnn_algo

Options d’optimisation cuDNN (paramètres).

Voir aussi

avx_math, clr_math, mkl_math, sse_math