Tensorflow GPU사용하기 No colab

딥러닝을 배우다보면 가장 많이 사용하게 되는게 Google Colab이다. 이유는 시간 제안이 있긴 하지만 그래도 무료로 GPU, TPU지원을 해주니까!!

 

그런데 GPU의 경우 지원되는 그래픽카드를 작창중인 피시 혹은 서버를 이용한다면 Colab을 사용하지 않아도 사용할 수 있다.

 

그런데 어떻게..? Colab에서는 메뉴 들어가서 GPU로 바꿔주기만 하면 됐는데..?라는 부분에서 글을 쓰고 요약은 바로 이거다.

import os
import tensorflow as tf 
from tensorflow.python.client import device_lib


# CPU, GPU확인 호출함수는 리스트형태로 값을 반환
# 보통은 0 - CPU / 1 - GPU으로 나옴
print(device_lib.list_local_devices() )


# CPU, GPU 사용 방법 둘 중 하나는 블락처리

os.environ["CUDA_VISIBLE_DEVICES"] = "0"	# GPU 사용

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"	# CPU 사용

 

위 방법에서 GPU가 보이지 않는다면 사용가능한 GPU가 있는지 확인하는 방법이 있다. 가끔 tensorflow에서 GPU를 인식 못하는 경우가 있는데 이떄 위 방법으로는 GPU가 안나온다. 아래 명령어를 실행하면 실행가능한 GPU가 있다면 출력된다.

! nvidia-smi

 

있는데 tensorflow에서 안나온다? 이러면  tensorflow자체는 설치가 되어 있는데 아마도 tensorflow-gpu 패키지가 설치가 안됐을 가능성이 크다. 두 개 버전이 같아야 인식이 가능하다고 하니 버전 맞추어서 설치해주면 된다.