cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

493 ワード


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']}