microsoftml.gpu_math: aceleración con NVidia CUDA
Uso
microsoftml.gpu_math(gpu_id: numbers.Real = -1,
cu_dnn: bool = False, cu_dnn_algo: str = 'ImplicitPrecompGemm')
Descripción
Implementación de CUDA de NVidia.
Argumentos
gpu_id
Identificador de dispositivo de GPU (configuración).
cu_dnn
Uso de cuDNN en GPU (configuración).
cu_dnn_algo
Opciones de optimización de cuDNN (configuración).
Vea también
Comentarios
Próximamente: A lo largo de 2024 iremos eliminando gradualmente GitHub Issues como mecanismo de comentarios sobre el contenido y lo sustituiremos por un nuevo sistema de comentarios. Para más información, vea: https://aka.ms/ContentUserFeedback.
Enviar y ver comentarios de