报错cuDNN: CUDNN_STATUS_EXECUTION_FAILED

加一句

torch.backends.cudnn.enabled = False

不行再查看cuda、python、pytorch、cudnn的版本兼容问题

猜你喜欢

转载自www.cnblogs.com/Manuel/p/12325409.html