output = input.matmul(weight.t()) RuntimeError: CUDA error: CUBLAS_STATUS_EXE...

Handsome_zhou

运行时报错：output = input.matmul(weight.t()) RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`

报错时的torch版本及cuda环境：pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
                                        google 之后没有找到合适的解决方案。

更换环境：pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html后解决。

歌者文明清理员

无

Handsome_zhou

歌者文明清理员 发表于 2023-6-26 15:51
发到“问题求助”！！！！！！！！！！！！！！！！！！！！！！！！！！！

我这个自己解决了，也要发问题求助那儿吗

歌者文明清理员

Handsome_zhou 发表于 2023-6-26 15:54
我这个自己解决了，也要发问题求助那儿吗

az，如果自己解决了，就不用了，抱歉看错了