Pytorch: 0.4.1
CUDA: 9.2
solution:
just add this line:
torch.backends.cudnn.benchmark = True
Link: https://stackoverflow.com/questions/62752522/runtimeerror-cudnn-error-cudnn-status-success
Pytorch: 0.4.1
CUDA: 9.2
solution:
just add this line:
torch.backends.cudnn.benchmark = True
Link: https://stackoverflow.com/questions/62752522/runtimeerror-cudnn-error-cudnn-status-success