解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

情况1:自己的cuda、cudnn、python、pytorch版本不匹配。注意,尤其是python的版本是很容易忽略的。

情况2(我的情况是这种):在情况1满足的条件下,查看自己的程序所使用的显卡占用量是不是满了,如果满了就换未使用的卡(‘cuda:3’ —> ‘cuda:0’)。

情况3:在代码中加入
torch.backends.cudnn.enabled = False

你可能感兴趣的:(Python技巧,python)