Python PyTorch가 GPU를 사용하는지 어떻게 확인할 수 있나요?, How do I check if PyTorch is using the GPU?

질문

PyTorch가 GPU를 사용하고 있는지 확인하는 방법은 무엇인가요? nvidia-smi 명령어는 GPU 활동을 감지할 수 있지만, 저는 파이썬 스크립트 내부에서 직접 확인하고 싶습니다.

답변

이러한 함수들이 도움이 될 것입니다:

>>> import torch

>>> torch.cuda.is_available()
True

>>> torch.cuda.device_count()
1

>>> torch.cuda.current_device()
0

>>> torch.cuda.device(0)
<torch.cuda.device at 0x7efce0b03be0>

>>> torch.cuda.get_device_name(0)
'GeForce GTX 950M'

이는 다음을 알려줍니다:

• CUDA가 사용 가능하며 한 장치에서 사용할 수 있습니다.
• Device 0은 GPU GeForce GTX 950M을 가리키며, 현재 PyTorch에서 선택되었습니다.