파이썬 GPU CUDA Torch를 이용한 확인

 

import torch


print(torch.cuda.is_available())


print(torch.cuda.current_device())


print(torch.cuda.get_device_name(0))

torch를 이용해서 gpu 확인 되면

아래 console로 확인됨.

True
0
NVIDIA A100 80GB PCIe