ycwsilent

您好，请问一下，你有实现吗？我在P4上面跑LOL数据集的测试，发现都报错 RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)` 似乎就是显卡不够；请问一下您是用的什么服务器和卡