Instantly share code, notes, and snippets.
add this line after import torch
torch.backends.cudnn.enabled = False