CUDA version of the project.

[0ystersauce]

Could you please tell me the CUDA version of the project cause I encounter the problem of "RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasGemmEx( handle, opa, opb, m, n, k, &falpha, a, CUDA_R_16F, lda, b, CUDA_R_16F, ldb, &fbeta, c, CUDA_R_16F, ldc, CUDA_R_32F, CUBLAS_GEMM_DFALT_TENSOR_OP)". I think the description of CUDA version may help.

[celsofranssa]

Hello, installing cuda version compatible with the installed torch seems to work properly.