사용
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