如果你遇到了这个错误 RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,
1. 首先确保所有版本都是匹配的,那么出现这个问题可能是因为你直接用pip install torch 指令了。
2. 那么,尝试用官网上的指令 (Pytorch官网找到合适的pytorch版本)重新安装pytorch,如下:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
如果你遇到了这个错误 RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,
1. 首先确保所有版本都是匹配的,那么出现这个问题可能是因为你直接用pip install torch 指令了。
2. 那么,尝试用官网上的指令 (Pytorch官网找到合适的pytorch版本)重新安装pytorch,如下:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch