Directorio de artículos
CUBLAS_STATUS_INVALID_VALUE
Informe de errores completo:
CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)`
A partir de ahora: hay tres posibilidades:
1. Tamaño del lote
Reducir el tamaño del lote. Mi tamaño de lote es originalmente 1, por lo que esto no debería ser un problema.
2. La versión no coincide
La versión CUDA y la versión cudnn no coinciden.
3. Discrepancia de dimensiones
Muchos de ellos se deben al problema de coincidencia del tamaño del tensor en la capa nn.Linner. Imprime el atributo de forma de dimensión del tensor. Obviamente, mi tamaño también es consistente con la operación de multiplicación.
3. Restaurar la configuración predeterminada de LD_LIBRARY_PATH
unset LD_LIBRARY_PATH
Luego vuelva a ejecutar el código. ¡Solución perfecta! ! !
unset LD_LIBRARY_PATHEs un LD_LIBRARY_PATHcomando que se utiliza para cancelar la configuración de variables de entorno. Normalmente LD_LIBRARY_PATHse utiliza para especificar la ruta de búsqueda de archivos de bibliotecas compartidas (como bibliotecas de vínculos dinámicos). Al configurar esta variable de entorno, le indica al sistema dónde buscar bibliotecas compartidas para que puedan cargarse en tiempo de ejecución.
Utilice unset LD_LIBRARY_PATHel comando para LD_LIBRARY_PATHeliminar el valor de la variable de entorno, lo que significa que el sistema utilizará la ruta de búsqueda de biblioteca predeterminada para encontrar archivos de biblioteca compartida. Esto es útil para solucionar algunos problemas relacionados con la configuración de la ruta de la biblioteca o restaurar la configuración predeterminada.
Tenga en cuenta que después de desarmar LD_LIBRARY_PATH, su sistema utilizará la ruta de búsqueda de biblioteca compartida predeterminada. Si anteriormente configuró LD_LIBRARY_PATHpara especificar una ruta específica, al desarmarla ya no se buscarán bibliotecas compartidas en esa ruta. Asegúrese de que su sistema aún pueda encontrar los archivos de la biblioteca compartida necesarios; de lo contrario, es posible que su aplicación no funcione correctamente.