Cuando probamos el modelo yolov5s.pt para el entrenamiento previo, el entrenamiento informará el error No se puede encontrar un algoritmo cuDNN válido para ejecutar la convolución. Cuando reinstale cudnn, el problema aún no se puede resolver.
Este problema es muy sencillo, no es que haya algún problema con nuestro cudnn. En cambio, dado que la memoria de video puede ser insuficiente durante el entrenamiento, solo necesita ajustar el tamaño del lote a un tamaño más pequeño en este momento.