Skip to content

Latest commit

 

History

History
74 lines (42 loc) · 1.01 KB

File metadata and controls

74 lines (42 loc) · 1.01 KB
title gpu_math: gpu_math
description NVidia CUDA implementation.
keywords neural network, math, gpu
author WilliamDAssafMSFT
ms.author wiassaf
ms.date 07/15/2019
ms.topic reference
ms.service sql
ms.subservice machine-learning-services
ms.assetid
ROBOTS
audience
ms.devlang Python
ms.reviewer
ms.suite
ms.tgt_pltfrm
ms.custom
monikerRange >=sql-server-2017||>=sql-server-linux-ver15

microsoftml.gpu_math: Acceleration with NVidia CUDA

Usage

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

Description

NVidia CUDA implementation.

Arguments

gpu_id

GPU device id (settings).

cu_dnn

Use cuDNN on GPU (settings).

cu_dnn_algo

cuDNN optimization options (settings).

See also

avx_math, clr_math, mkl_math, sse_math