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
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