cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

 

cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

 

好像是内存不够了：

 

dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize,
                                         shuffle=False, num_workers=16) for x in ['gallery','query']}

解决方法：

dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize, shuffle=False, num_workers=0) for x in ['gallery','query']}