1/25/2023

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

ex)

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