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 裝置識別碼 (設定)。
cu_dnn
在 GPU 上使用 cuDNN (設定)。
cu_dnn_algo
cuDNN 最佳化選項 (設定)。
另請參閱
意見反應
https://aka.ms/ContentUserFeedback。
即將登場:在 2024 年,我們將逐步淘汰 GitHub 問題作為內容的意見反應機制,並將它取代為新的意見反應系統。 如需詳細資訊,請參閱:提交並檢視相關的意見反應