Поделиться через

microsoftml.gpu_math: ускорение на базе Nvidia CUDA

  • Статья

Использование

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

Описание

Реализация NVidia CUDA.

Аргументы

gpu_id

Идентификатор устройства GPU (параметры).

cu_dnn

Использовать cuDNN в GPU (параметры).

cu_dnn_algo

Параметры оптимизации cuDNN (параметры).

См. также

avx_math, clr_math, mkl_math, sse_math