파이토치 gpu 작동 확인 방법

1. 코드 실행

import torch

# GPU 사용 가능 여부 확인
if torch.cuda.is_available():
    device = torch.device("cuda")
    print(f"GPU is available: {torch.cuda.get_device_name(0)}")
else:
    device = torch.device("cpu")
    print("GPU is not available, using CPU instead.")

위 코드를 실행하면

현재 gpu를 사용할 수 있는지 없는지 프린트되어 나옵니다. 파이토치 gpu 작동 확인 방법에 대해 알아보았습니다.