加一句

torch.backends.cudnn.enabled = False

 

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

相关文章: