GPU 사용 가능한지 확인하기

Tensorflow

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

만약 위에서 import 에러가 뜬다면 아래 코드를 활용해서 tensorflow를 재설치해본다.

pip install --upgrade pip
pip uninstall tensorflow
pip install tensorflow

CPU만 사용 가능하다면, 위 명령어의 결과에  CPU 관련 정보만 표시된다

그림1. CPU만 사용 가능한 상태

 

Pytorch

 

import torch

torch.cuda.is_available() # GPU 사용 확인

torch.cuda.get_device_name(0) # 사용가능한 GPU 개수 반환

torch.cuda.device_count() # 환경 변수에 의해 사용가능한 GPU