RuntimeError: CUDNN_STATUS_MAPPING_ERROR in prep_binary_masks
Ankit050 opened this issue · 1 comments
Ankit050 commented
Hi, I'm facing an issue in running the prep_binary_masks. My PyTorch version is 1.2.0 and torchvision is 0.4.0. CUDA version is 10.0.130.
The error occurs at the line:
preds = fa.get_landmarks(x)
Has someone faced a similar issue?
Ankit050 commented
Update: It works by updating Pytorch to 1.3.0 and torchvision to 0.4.1