[python] tensorflow에서 GPU 사용가능 여부 확인하기

프로그래밍/python|2020. 5. 12. 17:54| bskyvision 심교훈

GPU를 사용하면 빠르죠. 왜냐하면 병렬 연산이 가능하기 때문입니다. 이것이 무슨 말인가 하면 명령을 하나씩 순차적으로 처리하는 CPU와 달리 GPU는 여러 명령을 동시에 처리할 수 있다는 뜻입니다. 그래서 연산량이 많은 딥러닝을 다룰 때는 GPU를 사용하는 것이 좋습니다. 

 

오늘은 tensorflow에서 GPU를 사용 가능 여부를 확인해보는 것에 대해 다루도록 하겠습니다. 사용가능 여부를 알기 위해서는 다음 두 줄의 코드를 입력해보면 됩니다.

 

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

만약 다음과 같이 CPU 관련된 내용 뿐만 아니라 GPU에 관한 것이 출력된다면 사용가능한 것입니다. 

 

 

GPU 사용가능

 

 

그런데 다음과 같이 이와 달리 CPU에 관한 것만 나온다면 GPU는 사용할 수 없는 상태입니다. 

 

 

CPU만 사용가능

 
참고로 GPU를 사용하시려면 컴퓨터에 그래픽 카드가 깔려 있어야 하고, 또 그에 맞는 드라이버와 적절한 버전의 CUDA, cuDNN이 설치되어 있어야합니다.

 

 

 

<참고자료>

[1] https://light-tree.tistory.com/25, All about, "CPU와 GPU의 차이"

[2] https://lucycle.tistory.com/237, LuCycle, "텐서플로우 GPU 작동 확인하는 방법"

  1. BlogIcon DNA구너 2020.05.12 21:08 신고 댓글주소  수정/삭제  댓글쓰기

    엔비디아 글카를 사야 하는 이유가 여기있죠..ㅎㅎ