search

yfeng997 / MadMario

Interactive tutorial to build a learning Mario, for first-time RL learners
192 stars 68 forks source link

run main.py ,some errors happen. what the GPU environments of your computer? #14

Open greenwilddog opened 2 years ago

greenwilddog commented 2 years ago

when i run main.py. it shows that RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc) how to solve this error?

Whit3Snow commented 1 year ago

I meet the same question as you,have you solve it?

mfirmin commented 1 year ago

I ran into the same issue. For me, upgrading pytorch and cuda to the latest versions (1.13.1 and 11.6 respectively at the time of this comment) fixed it for me.

pip install --upgrade torch torchvision torchaudio updated both pytorch and cuda.