Badminton-Sport-Analysis-Computer-Vision
Badminton-Sport-Analysis-Computer-Vision copied to clipboard
error cudnn
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