microsoftml.gpu_math: Ускорение с NVidia CUDA

Usage

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

Description

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

Arguments

gpu_id

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

cu_dnn

Используйте cuDNN на GPU (настройки).

cu_dnn_algo

Настройки оптимизации cuDNN.