Pytorch报错:cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

如果你遇到了这个错误 RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,

1. 首先确保所有版本都是匹配的,那么出现这个问题可能是因为你直接用pip install torch 指令了。

2. 那么,尝试用官网上的指令 (Pytorch官网找到合适的pytorch版本)重新安装pytorch,如下:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

 


版权声明:本文为tel_Annie原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。