Solucionado
La configuración de CPU's, GPU's y TPU's en google colab ha cambiado desde la publicación de este curso y ahora la única configuración disponible es T4 GPU. Supondría que habría una mejora en los tiempos de ejecución del código torch pero no he conseguido mejores tiempos de ejecución comparado con el código numpy. Alguna explicación?
Hola, Gabriel! Cómo va?
El uso de una GPU como la T4 en Google Colab puede no resultar en una mejora significativa de rendimiento si las tareas no están optimizadas para aprovechar el procesamiento paralelo que ofrecen las GPUs. Las GPUs son más efectivas para grandes volúmenes de datos y cálculos intensivos que pueden paralelizarse. Si las operaciones son pequeñas o el código no está diseñado para ejecución paralela, como puede ser el caso con algunas funciones de Numpy, la GPU no acelerará el rendimiento. Además, asegúrate de que tu código esté efectivamente utilizando la GPU configurando correctamente el entorno y las operaciones para ejecutar en la GPU. ;)
Abrazos y buenos estudios!
Si este post te ayudó, por favor, marca como solucionado ✓.