[pytorch] RuntimeError: FIND was unable to find an engine to execute this computation

(본 포스팅은 벨로그의 글을 옮겨온 것입니다.)

 

포팅하다가 같은 런타임에러가 여러 코드에서 동시에 일어나서 검색해보니 pytorch와 cuda 버전이 서로 안맞는 오류라고 한다. 여러 번 anaconda 가상환경을 새로 만들어 시도해봐도 오류를 해결할 수 없어서 결국 파이토치를 지웠다가 깔려고 했는데 연관된 다른 계정들이 많아서 고민하고 있던 찰나 도움을 받아 해결했다.

해결방법은 간단하게 아나콘다 가상환경 내에 cuda를 독자적으로 새 버전을 설치해주는것.
버전과 환경은 파이토치 사이트에서 쉽게 클릭으로 조정된다.

pytorch 사이트 버전 바로가기

터미널 창에 가상환경이 구동된 채로 아래 코드를 실행했다.

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

듣기로는 pytorch-cuda=11.7이 부분이 가상환경 내에 설치하는 거라고 한다.

아무튼 이 문제는 해결..