runtime error: CUDNN_STATUS_MAPPING_ERROR

[HIWFS]

When I execute “prediction = F.sigmoid(model(input_img)).data[0].cpu().numpy()” is reporting error “runtime error: CUDNN_STATUS_MAPPING_ERROR”，what should I do ? thank you very much!

[Saransh Karira]

its using PyTorch 0.4, try instead:
prediction = torch.sigmoid(model(input_img)).item()