import torch
torch.cuda.is_available()
True
torch.cuda.current_device()
0
torch.cuda.get_device_name(0)
'GeForce TITAN X'
'실습 > pytorch' 카테고리의 다른 글
| load_state_dict 성능 문제, [nn.ModuleList] (2) | 2018.11.08 |
|---|---|
| Pytorch eval 성능 문제 (0) | 2018.11.08 |
| pytorch 메모 (0) | 2018.03.11 |