tip
cuda v10.1 getDevice() 오류 해결하기
mtoc
2020. 4. 15. 15:47
아니 외않되?
안 되게 해놨으니 안 되는 거다...
이 오류는 적어도 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 버전을 꼭 확인하자!