슬기로운 연구생활

[ GPU ] CUDNN_STATUS_EXECUTION_FAILED 본문

슬기로운 에러 생활

[ 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
Comments