解决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


版权声明:本文为qq_36321330原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。