问题:
调用显卡时,出现RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
问题分析:
出现这种问题,一般是因为cuda、cudnn、显卡驱动、torch版本不匹配造成的。
解决:
1、第一个办法就是重启大法哈哈哈;
2、查看cuda版本与cudnn版本是否匹配;
3、查看pytorch版本是否与显卡匹配,不知道是否匹配,就试试,或许降低版本会有意料之外的效果,即解决了;
4、看驱动版本是否与cuda匹配;
我的解决办法:环境10.2,驱动 440.33.01,torch1.8,一直才出现该问题,后来降低版本到1.7就可以了。