Handsome_zhou 发表于 2023-6-26 15:48:37

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

运行时报错: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后解决。

歌者文明清理员 发表于 2023-6-26 15:51:31

本帖最后由 歌者文明清理员 于 2023-6-26 15:55 编辑

Handsome_zhou 发表于 2023-6-26 15:54:05

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

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

歌者文明清理员 发表于 2023-6-26 15:55:48

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

az,如果自己解决了,就不用了,抱歉看错了
页: [1]
查看完整版本: output = input.matmul(weight.t()) RuntimeError: CUDA error: CUBLAS_STATUS_EXE...