Pytorch GPU Check

Pytorch GPU Check

Pytorch GPU Check에 가장 대표적인 것들만 모아봤습니다.

torch.cuda.current_device()

torch.cuda.device(0)

torch.cuda.device_count()

torch.cuda.get_device_name(0)

torch.cuda.is_available()

개인적으로 이 중 가장 중요한 것은 아래 3가지 일 것이라 생각합니다.

torch.cuda.is_available() # cuda가 사용 가능한지 확인

torch.cuda.device(device) #device를 변경할 때 사용

torch.cuda.get_device_name(0) # device 이름을 return 합니다.

여러분들이 직접 실행해보시고 확인을 통해 이해하시는 것을 추천드립니다.

- 참고로 위 API documents는 아래 2번 URL에 적어놨습니다.

1. 

가장 잘 설명해주고 있는 URL은 아래 stackoverflow에 올라와있는 질문을 통해서인 듯 합니다.

https://stackoverflow.com/questions/48152674/how-to-check-if-pytorch-is-using-the-gpu

2. Pytorch Cuda 관련 documents URL

https://pytorch.org/docs/stable/cuda.html

감사합니다.~