why CUDA error？
Closed this issue · 4 comments
viko-3 commented
python train.py --task bindingdb
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
NVIDIA-SMI 455.45.01 Driver Version: 455.45.01 CUDA Version: 11.1
kexinhuang12345 commented
it seems to be a cuda issue; not a moltrans issue?
viko-3 commented
it seems to be a cuda issue; not a moltrans issue?
what is your cuda version?
kexinhuang12345 commented
Mine is 10.2
kexinhuang12345 commented
closing for now