python运行报错RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

python运行报错RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

问了chatgpt、new bing，得到的答复是，cuda版本不对，要不然就是pytorch的版本不对。

所以试了一下都不对。最后试了一下embedding维度，是不是太小了。调大后正常运行。
只能说，gpu和cpu报错还不一样，之前用cpu曝出来错误是能看懂啥意思的英语。