Pytorch에서 Device Check(GPU)

pytorch를 이용하면서 해당 장비에서 gpu device를 체크하기 위한 방법

 

import torch

torch.cuda.is_available()

반환값 = True of False

 

GPU 모델을 알기 위해서는?

torch.cuda.get_device_name(0) # 0 is gpu device id

반환값 = GPU 모델 이름(예: GeForch RTX 2080)

 

GPU가 몇개인지 알려면? 

torch.cuda.device_count()

반환값 = GPU 수