Usage
microsoftml.gpu_math(gpu_id: numbers.Real = -1,
cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')
Description
NVidia CUDA 实现。
Arguments
gpu_id
显卡设备ID(设置)。
cu_dnn
在GPU上使用cuDNN(设置)。
cu_dnn_algo
cuDNN 优化选项(设置)。
microsoftml.gpu_math(gpu_id: numbers.Real = -1,
cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')
NVidia CUDA 实现。
显卡设备ID(设置)。
在GPU上使用cuDNN(设置)。
cuDNN 优化选项(设置)。