더북(TheBook)

부록 A
텐서플로 GPU 버전 설치하기

 

 

부록 A에서는 텐서플로 GPU 버전을 설치하는 과정을 설명하였다. 텐서플로의 성능을 잘 활용하려면 아무래도 GPU 버전을 사용하는 것이 좋다. GPU 버전을 잘 사용하기 위해서는 NVIDIA 계열의 그래픽 카드가 필요하며, 리눅스 OS에서 사용하는 것이 좋지만, 여기서는 보다 일반적인 윈도의 GPU가 설치된 환경을 기준으로 소개하고자 한다. 여기에서 소개된 내용은 일반적인 환경에서 설치하는 방법이지만, 설치 시 PC의 환경마다 각기 다른 상황이 발생할 수도 있으니 공식 사이트를 통해 최신 정보를 자주 확인하기 바란다.

GPU 버전을 설치할 때 가장 주의할 점은 바로 설치할 프로그램의 버전을 맞추는 것이다. 설치되는 프로그램은 CUDA, cuDNN, tensorflow-GPU 버전이며 각각은 OS, CUDA Toolkit 등의 버전이 맞아야 한다. 특히, 사용 중인 그래픽 카드가 CUDA를 지원하는지를 다음 주소에서 먼저 확인해야 한다.

URL https://developer.nvidia.com/cuda-gpus

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.