microsoftml.gpu_math: ускорение на базе Nvidia CUDA
Использование
microsoftml.gpu_math(gpu_id: numbers.Real = -1,
cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')
Описание
Реализация NVidia CUDA.
Аргументы
gpu_id
Идентификатор устройства GPU (параметры).
cu_dnn
Использовать cuDNN в GPU (параметры).
cu_dnn_algo
Параметры оптимизации cuDNN (параметры).
См. также
Обратная связь
Ожидается в ближайшее время: в течение 2024 года мы постепенно откажемся от GitHub Issues как механизма обратной связи для контента и заменим его новой системой обратной связи. Дополнительные сведения см. в разделе https://aka.ms/ContentUserFeedback.
Отправить и просмотреть отзыв по