问题起因:
在运行论文代码时出现了RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 报错。
解决方法:
如果不是cuda、cudnn配置的问题,那有可能是电脑的线程数不满足所配置的参数,在更改num_workers参数后问题得到了解决。
本人电脑是8线程,将num_workers调至小于8即可。
查找电脑线程的方法:
- win+R;
- 输入cmd,确定;
- 输入 wmic,回车;
- 输入 cpu get numberOfLogicalProcessors,回车。
参考:(36条消息) 【Pytorch】RuntimeError: CUDA out of memory 问题解决_柠檬啵的博客-CSDN博客
标签:STATUS,cuDNN,RuntimeError,CUDNN,Pytorch,线程 From: https://www.cnblogs.com/zc-030/p/17210762.html