다음을 통해 공유


microsoftml.gpu_math: NVidia CUDA를 사용한 가속

사용

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

Description

NVidia CUDA 구현입니다.

인수

gpu_id

GPU 디바이스 ID(설정)입니다.

cu_dnn

GPU에서 cuDNN을 사용(설정)합니다.

cu_dnn_algo

cuDNN 최적화 옵션(설정)입니다.

추가 정보

avx_math, clr_math, mkl_math, sse_math