1In [1]: import torch
2
3In [2]: torch.cuda.current_device()
4Out[2]: 0
5
6In [3]: torch.cuda.device(0)
7Out[3]: <torch.cuda.device at 0x7efce0b03be0>
8
9In [4]: torch.cuda.device_count()
10Out[4]: 1
11
12In [5]: torch.cuda.get_device_name(0)
13Out[5]: 'GeForce GTX 950M'
14
15In [6]: torch.cuda.is_available()
16Out[6]: True
17