Open Lopa07 opened 11 months ago
Could you provide more details, e.g., where/which line does this error occur in the project?
I am getting the same issue. Tried different cuda versions no luck. Were you @Lopa07 able to fix it? Thanks!
I was able to fix this by using cuda-toolkit 10.2 with cudnn8.7 for cuda10.2 (https://developer.nvidia.com/rdp/cudnn-archive#a-collapse870-102). Hope this helps. @Lopa07
The following version works as well as long as the appropriate cudann version is installed.
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
This guide is helpful for cudann installation.
Getting the error,
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling 'cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)'
while trying the following training command:I am using python 3.7, CUDA 10.1.