파이썬 GPU CUDA Torch를 이용한 확인
IT/JAVA
2023. 8. 23. 16:58
728x90
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