0 votos

Cómo activar la GPU en Mac M1 para YOLOv5 (usando TensorFlow)

Estoy usando TensorFlow para aprendizaje automático, pero el problema es que la GPU no se activa en la sesión mientras se realiza la parte de entrenamiento. Sólo utiliza la CPU y teniendo un M1 con motores neuronales en él quiero utilizar toda la potencia de procesamiento del Mac. Me gustaría conseguir una solución rápida para ello.

La memoria asignada a la GPU es muy pequeña y para entrenar un conjunto de datos debería producirse una alta asignación de GPU; cosa que no está ocurriendo.

Screenshot of several graphs showing different statistics for the GPU

0voto

siva Puntos 23

Instalar el plug-in tensorflow-metal

python -m pip install tensorflow-metal

https://developer.apple.com/metal/tensorflow-plugin/

AppleAyuda.com

AppleAyuda es una comunidad de usuarios de los productos de Apple en la que puedes resolver tus problemas y dudas.
Puedes consultar las preguntas de otros usuarios, hacer tus propias preguntas o resolver las de los demás.

Powered by:

X