报错cuDNN: CUDNN_STATUS_EXECUTION_FAILED

加一句

torch.backends.cudnn.enabled = False

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

posted on 2020-02-18 11:33 Manuel 阅读(1507) 评论(0) 编辑收藏举报

会员力量，点亮园子希望

刷新页面返回顶部