配置: Python 3.7.6, CUDA 10.0,Pytorch 1.3.1, cudNN 7.6
运行时出现如题错误
解决办法: 在代码前面添加torch.backends.cudnn.enabled = False
作用: 网传是禁用cuDNN作用,官网没有查到相应API
说明: 网上查到的靠谱的解决办法,意思大都是说各个版本匹配的问题,个人觉得有一定道理,可能是这么回事,以上办法,也只是针对个人的环境下,可以解决
ps:
1、什么是CUDA
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
2、什么是CUDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。
3、CUDA与CUDNN的关系
CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUD