RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED解决办法
配置: Python 3.7.6, CUDA 10.0,Pytorch 1.3.1, cudNN 7.6运行时出现如题错误解决办法: 在代码前面添加torch.backends.cudnn.enabled = False作用: 网传是禁用cuDNN作用,官网没有查到相应API说明: 网上查到的靠谱的解决办法,意思大都是说各个版本匹配的问题,个人觉得有一定道理,可能是这么回事,以上办法,也只是针对个人的环境下,可以解决ps:1、什么是CUDACUDA(ComputeUnified Device
原创
2020-09-27 12:24:55 ·
2237 阅读 ·
1 评论