解决报错:CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemmStridedBatched( handle, opa, opb, m......

CUBLAS_STATUS_INVALID_VALUE

完整报错:

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)`

就目前来看:有以下三种可能:

1. Batchsize

将Batchsize调小,我的batchsize本来就是1,所以应该不是这个问题。

2. 版本不匹配

CUDA版本和cudnn版本不匹配。

3. 维度不匹配

很多是由于nn.Linner层的 Tensor 尺寸匹配问题,打印下张量的维shape属性,显然我的尺寸也是符合乘法操作的。

3. 恢复 LD_LIBRARY_PATH 默认设置

unset LD_LIBRARY_PATH

然后重新运行代码。完美解决!!!

unset LD_LIBRARY_PATH 是一个用于取消环境变量 LD_LIBRARY_PATH 设置的命令。通常情况下,LD_LIBRARY_PATH 用于指定共享库文件(例如动态链接库)的搜索路径。通过设置这个环境变量,您可以告诉系统在哪里查找共享库,以便在运行时加载它们。

使用 unset LD_LIBRARY_PATH 命令可以将 LD_LIBRARY_PATH 环境变量的值删除,这意味着系统将使用默认的库搜索路径来查找共享库文件。这对于解决一些与库路径设置相关的问题或恢复默认设置非常有用。

请注意,在取消设置 LD_LIBRARY_PATH 后,您的系统将使用默认的共享库搜索路径。如果您之前设置了 LD_LIBRARY_PATH 以指定特定路径,取消设置后,系统将不再搜索该路径中的共享库。确保您的系统仍然能够找到所需的共享库文件,否则可能会导致应用程序无法正常工作。

  • 7
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 7
    评论
"runtimeerror: cuda error: cublas_status_invalid_value when calling `cublassg` 是在调用 `cublassg` 函数时发生的 CUDA 错误,错误代码为 `cublas_status_invalid_value`。这个错误通常是由于输入参数的值无效导致的。 首先,要解决这个错误,我们需要确定导致错误的原因。一种可能性是输入参数超出了有效范围或具有无效的值。另一种可能性是输入参数的大小与函数预期的大小不匹配。 为了解决这个错误,我们可以采取以下步骤: 1. 检查输入参数的值是否满足函数的要求。查看函数文档以确定每个参数的有效值范围,并确保输入满足这些要求。 2. 检查输入参数的大小是否正确。确保输入参数的维度与函数期望的维度匹配。 3. 检查输入参数是否进行了正确的初始化。如果输入参数没有正确初始化,可能会导致无效的值。 另外,检查 CUDA 版本和驱动程序版本是否与函数兼容也很重要。某些 CUDA 函数需要特定的驱动程序版本才能正常工作。 最后,如果以上方法无法解决问题,可以尝试在 CUDA 环境中调试并查找错误的根本原因。使用调试工具来检查内核的执行方式,并查看可能出现的错误。 总之,`runtimeerror: cuda error: cublas_status_invalid_value when calling `cublassg` 是一个 CUDA 错误,表示在调用 `cublassg` 函数时发生了值无效的情况。通过检查输入参数的值、大小和初始化情况,以及检查 CUDA 和驱动程序的兼容性,可以解决这个问题。如果问题仍然存在,可以使用调试工具来进一步分析和解决错误。"
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

莫余

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值