pytorch cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

1735 ワード

解決方法1.テスト
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