GPU 사용하게 하기

텐서 플로우를 돌리는데 있어 GPU가 있고 없고는 성능에 많은 차이를 보입니다. 

 

해서, 사용하고 있는 PC/노트북의 GPU가 텐서 플로우를 지원한다면, 가능하게 설정해주는 것이 좋습니다.

 

순서는,

• PC/노트북에 있는 그래프카드가 GPU 지원하는지 확인
• 최신 GPU 드라이버로 업그레이드
• GPU Toolkit 설치

그래프카드가 GPU 지원하는지 확인하기

'내컴퓨터/속성/장치관리자/디스플레이 어댑터'에서 NVIDIA 그래픽 카드 모델 번호 확인

 

 

 

TensorFlow에서 GPU를 사용하려면 NVIDIA의 CUDA 기술을 사용한 GPU카드라야하고, CUDA기술을 사용한 카드인지는 여기를 가보면 알 수 있습니다.

 

 

내꺼는 NVIDIA GeForce GT 650M이고 다행히 CUDA기술을 사용하고 있는 것

https://www.geforce.com/hardware/technology/cuda/supported-gpus?field_gpu_type_value=notebook-gpus&page=2

 

최신 GPU 드라이버 설치하기

최신 드라이버를 설치해야합니다. 

드라이버 다운로드는 여기

 

 

다운로드한 파일을 실행해서 설치합니다.

 

CUDA Toolkit 설치하기

텐서 플로우 버전마다 지원하는 CUDA Toolkit 버전이 다릅니다. 

 

현재, 텐서 플로우는 v2.1이 최신이고, 이 버전에는 CUDA 10.1을 지원합니다. (CUDA 최신 버전은 10.2 이기에 10.2를 받은 실수를 할 수 있는데, 그러지 않고 10.1을 받아서 설치해야합니다. )

(이러한 사항은 여기 텐서 플로우 사이트에 설명되어 있습니다.)

 

CUDA Toolkit 10.1을 다운로드 할 수 있는 곳은 여기.

 

다운로드한 파일을 실행해서 설치합니다.

 

설치가 다 된 후, 잘 되었는지 확인은 cmd 창에서 `nvcc -V`로 확인

 

만약 제대로 동작하지 않으면 CUDA Toolkit 설치 가이드 페이지를 참조해서 확인.

 

 

-끝-