ERROR 记录 : No se pudo obtener el algoritmo de convolución. Esto probablemente se deba a que cuDNN no se pudo inicializar.

Error de inicialización de GPU:
con pycharm, solo se puede ejecutar un programa con GPU. Primero detenga y vuelva a ejecutar otros programas que no se ejecutan. Si hay errores, agregue el siguiente código:

import os
import tensorflow as tf

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
os.environ['TF_FORCE_GPU.ALLOW_GROWTH'] = 'TRUE'
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

IndexError: el índice 60 está fuera de los límites para el eje 1 con
un error de tamaño 52 YOLOv3, esta es la posición de la etiqueta no corresponde al tamaño de la imagen, por ejemplo, la etiqueta es 1256,670,1308,760,0, pero el tamaño real de la imagen es solo 1100 * 1100 ...

Supongo que te gusta

Origin blog.csdn.net/weixin_45371989/article/details/108738745
Recomendado
Clasificación