microsoftml.gpu_math: Acceleration med NVidia CUDA

Usage

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

Description

NVidia CUDA-implementering.

Arguments

gpu_id

GPU-enhets-id (inställningar).

cu_dnn

Använd cuDNN på GPU:n (inställningar).

cu_dnn_algo

cuDNN-optimeringsalternativ (inställningar).

Relaterat innehåll

• microsoftml.avx_math: Acceleration med AVX-instruktioner
• microsoftml.clr_math: Acceleration med .NET-matematik
• microsoftml.mkl_math: Acceleration med Intel MKL
• microsoftml.sse_math: Acceleration med SSE-instruktioner