【Pytorch】RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 问题解决

问题起因:

  在运行论文代码时出现了RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 报错。


 

解决方法:

如果不是cuda、cudnn配置的问题,那有可能是电脑的线程数不满足所配置的参数,在更改num_workers参数后问题得到了解决。

本人电脑是8线程,将num_workers调至小于8即可。

查找电脑线程的方法:

  1. win+R;
  2. 输入cmd,确定;
  3. 输入 wmic,回车;
  4. 输入 cpu get numberOfLogicalProcessors,回车。

     

 

 

 


 

参考:(36条消息) 【Pytorch】RuntimeError: CUDA out of memory 问题解决_柠檬啵的博客-CSDN博客

posted @ 2023-03-13 11:31  盏茶  阅读(2993)  评论(0)    收藏  举报