실습/pytorch
PyTorch GPU check
gldmg
2018. 8. 29. 10:47
import torch
torch.cuda.is_available()
True
torch.cuda.current_device()
0
torch.cuda.get_device_name(0)
'GeForce TITAN X'