본문 바로가기
tip

cuda v10.1 getDevice() 오류 해결하기

by mtoc 2020. 4. 15.

아니 외않되?

안 되게 해놨으니 안 되는 거다...

이 오류는 적어도 cuda이 성공적으로 설치되고, cuDNN의 버전이 cuda와 일치하며 환경변수 설정이 잘 이루어졌을 경우에 나타난다.
원인은 간단하다. 해당 버전의 cuda를 컴에 깔려있는 tensorflow 버전이 지원하지 않기 때문이다.

tensorflow GPU 지원 웹사이트(https://www.tensorflow.org/install/gpu?hl=ko)에 들어가면 다음과 같은 항목을 확인할 수 있다.

10.1 이상 버전 해결법

이미 텐서플로가 설치되어 있다면

pip freeze

위 코드로 텐서플로 버전을 확인해본다.
만약 2.1.0 이하라면 다음 코드를 쉘에 입력

pip uninstall tensorflow-gpu
pip install tensorflow-gpu==2.1.0

그리고나서도 오류가 뜨면 jupyter notebook을 다시 실행시켜주거나, 컴퓨터 재부팅하면 잘 돌아가는 것을 볼 수 있다.
이거 해결하고 암이 나았습니다... 정말로 ㅠ

쿠다 설치할 때는 꼭 공식 홈페이지에서 cuda, cuDNN, tensorflow 버전을 꼭 확인하자!

댓글