cuDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方案

原因1:cuda和pytorch版本不一致

检查版本是否一致:

python3
import torch
import torchvision
print(torch.cuda.is_available())

运行结果:
cuDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方案_第1张图片

如果是False,则去pytorch官网下载对应版本。例如cuda11.3+pytorch11.0:

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

原因2:不知名原因

在代码中import torch后加入:

torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True

还不行就把第一条换成False。(原理我也不知道,有时候有用hhh)


原因3:设备和不支持当前cuda版本

我采用A100-SXM4-80GB * 1卡训练时用cuda10.1又慢又报错,重装了cuda11.3就解决了


原因4:显卡不够

换个更大(贵)的卡

你可能感兴趣的:(深度学习,pytorch,python)