Hi,
when using get_segmentation_learner function with architecture_name="unet" and backbone_name="resnet50" I'm getting "RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED" error. I'm using torch 1.7.1, torchvision 0.8.2 and cuda 10.2 because of fastai compatibility.
Do you have some idea why this happens ?
Hi,
when using get_segmentation_learner function with architecture_name="unet" and backbone_name="resnet50" I'm getting "RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED" error. I'm using torch 1.7.1, torchvision 0.8.2 and cuda 10.2 because of fastai compatibility.
Do you have some idea why this happens ?