pytorch cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
1735 ワード
解決方法1.テスト
2.モデルを実行する前にコードを追加
最終OK
import torch
from torch.backends import cudnn
#
a = torch.tensor(1.)
#
a.cuda()
#
tensor(1., device='cuda:0')
#
cudnn.is_available()
# True
cudnn.is_acceptable(a.cuda())
# true
2.モデルを実行する前にコードを追加
A cudnn
torch.backends.cudnn.enabled = False
B
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
最終OK