关于#python#的问题，如何解决？

python=3.7
torch=1.2.0
torchvision=0.4.0
cuda=10.0
GPU=RTX3060  6G
CPU=i7-12700H

用CPU跑没问题，用GPU跑时程序的loss反向传播时报错，求解答

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

pytorch、cuda和python版本匹不匹配 用gpu跑的时候所有的版本都得匹配，不匹配就报错