使用量
microsoftml.gpu_math(gpu_id: numbers.Real = -1,
cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')
說明
NVidia CUDA 實作。
論點
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 優化選項(設定)。