Condividi tramite


microsoftml.gpu_math: accelerazione con NVidia CUDA

Utilizzo

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

Descrizione

Implementazione di NVidia CUDA.

Argomenti

gpu_id

ID dispositivo GPU (impostazioni).

cu_dnn

Usare cuDNN su GPU (impostazioni).

cu_dnn_algo

Opzioni di ottimizzazione cuDNN (impostazioni).

Vedi anche

avx_math, clr_math, mkl_math, sse_math