如果你遇到了这个错误 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版权协议,转载请附上原文出处链接和本声明。