슬기로운 에러 생활
[ GPU ] CUDNN_STATUS_EXECUTION_FAILED
vhrehfdl
2020. 5. 24. 14:53
* Error Message : RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
* Problem
pytorch의 GPU가 문제가 발생했다.
최근의 GPU를 2080TI로 변경했는데 이게 문제였다.
GPU 2080TI로 하고 CUDA 9점대 버전을 사용하면 Pytorch가 제대로 인식을 못하는 에러가 발생한다.
* Solution
해결방법은 간단한다.
1. CUDA를 10점대 버전으로 업데이트 한다.
2. Pytorch를 업데이트 한다.
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch