Badminton-Sport-Analysis-Computer-Vision icon indicating copy to clipboard operation
Badminton-Sport-Analysis-Computer-Vision copied to clipboard

error cudnn

Open yosuaalvin opened this issue 5 months ago • 0 comments

when i running densenet in pytorch there is an error /media/yosua/NewVolume/badminton-env/lib/python3.7/site-packages/torch/nn/functional.py in batch_norm(input, running_mean, running_var, weight, bias, training, momentum, eps) 1921 return torch.batch_norm( 1922 input, weight, bias, running_mean, running_var, -> 1923 training, momentum, eps, torch.backends.cudnn.enabled 1924 ) 1925

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED how to resolve that? i have use cudnn version 7.6.5 and cuda 10.1

yosuaalvin avatar Jan 06 '24 06:01 yosuaalvin