/Python/How to identify GPU with pytorch script

How to identify GPU with pytorch script

In [1]: import torch

In [2]: torch.cuda.current_device()
Out[2]: 0

In [3]: torch.cuda.device(0)
Out[3]: <torch.cuda.device at 0x7efce0b03be0>

In [4]: torch.cuda.device_count()
Out[4]: 1

In [5]: torch.cuda.get_device_name(0)
Out[5]: 'GeForce GTX 950M'

In [6]: torch.cuda.is_available()
Out[6]: True

Source: stackoverflow.com

Similar codes

1 vote179 views1 answer
Get last year of today python
Python
24 votes169 views2 answers
Find python
Python
0 vote190 views1 answer
Python split string keep delimiter
Python
0 vote171 views1 answer
Turn off warning when import python
Python
16 votes169 views4 answers
Sort tuple list python
Python
0 vote213 views1 answer
Read tar.gz file python
Python
0 vote186 views1 answer
Check how many letters in a string python
Python
0 vote182 views2 answers
Extra error
Python
0 vote171 views2 answers
How to plot two different class in different colour in python
Python
-1 vote319 views2 answers
Python virus
Python