CUDNN_STATUS_EXECUTION_FAILED

Pytorch가 GPU를 제대로 인식하지 못해 발생한 에러이다.
해결방법은 간단한다. CUDA 버전과 Pytorch를 업데이트 한다.

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html