RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`


I was suffering above (title) error during few hours.

The reason is wrong cuda version installed with pytorch.

My cuda version is 11.6, but install version is 11.7.

So I print it -> "torch.__version__"

It returend -> "1.13.1+cu117"

You can check your cuda version using this command

> nvidia-smi

so, remove all torch, torchvision packaged

> pip uninstall torch torchvision

and install again


pip3 install torch torchvision torchaudio --extra-index-url \https://download.pytorch.org/whl/cu116

Then it works well.

Thank you.


No comments:

Post a Comment