运行Torch代码时,遇到错误RuntimeError: CUDA error: 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)
。
报错位置是由于 torch.bmm()
这个函数引起的,查阅中文论坛,很多是由于nn.Linner
层的 Tensor 尺寸匹配问题,而我这边的尺寸是没有问题的。
PS: 这个代码使用 Pycharm 可以完美运行,但是在 MobaXterm ssh remote 时会报错。
github issue
在linux 命令行输入以下命令
unset LD_LIBRARY_PATH
该命令作用:
unset LD_LIBRARY_PATH 是一个用于清空 Linux 或 Unix 操作系统环境变量 LD_LIBRARY_PATH 的命令。
在 Linux 或 Unix 系统中,LD_LIBRARY_PATH 是一个环境变量,用于指定共享库文件(也称为动态链接库)的搜索路径。当一个程序启动时,系统会在 LD_LIBRARY_PATH 指定的路径下搜索共享库文件。如果 LD_LIBRARY_PATH 没有被设置,系统会使用默认的共享库搜索路径。
使用 unset LD_LIBRARY_PATH 命令可以清空 LD_LIBRARY_PATH 环境变量,这意味着系统将只使用默认的共享库搜索路径。这个命令通常用于解决程序运行时由于错误的 LD_LIBRARY_PATH 设置导致的问题。
至此,本机该问题已解决,也算是为中文论坛提供一种解决该问题的方法。