check_gpu.py

import torch
print("PyTorch :", torch.__version__)
print("CUDA    :", torch.version.cuda)
print("GPU?    :", torch.cuda.is_available())
if torch.cuda.is_available():
    print("GPU name:", torch.cuda.get_device_name(0))
else:
    print("GPU name: 없음")