Hi I have this error of :
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
I am just trying to follow what is in this tutorial:
https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html 25
It works fine on my machine:!
Screen Shot 2021-02-23 at 9.38.24 PM|621x500 16
but there is an error when I do the same on the server:
Screen Shot 2021-02-23 at 9.38.44 PM1830×1198 324 KB
I would appreciate it if someone can help! Thanks