/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

3 votes301 views1 answer
Knowing the sum of null value is pandas dataframe
Python
0 vote364 views1 answer
How to change the colour of axes in matplotlin
Python
0 vote362 views2 answers
User logout in django rest framework
Python
4 votes352 views6 answers
Python counter
Python
1 vote322 views4 answers
Pandas list comprehension
Python
0 vote411 views1 answer
Flask session auto logout in 5 mins
Python Flask
2 votes319 views1 answer
Python argparser flags
Python
0 vote356 views1 answer
How to change defaul tpython version\
Python
0 vote329 views1 answer
Python Record live streams (TS FILES)
Python
4 votes354 views2 answers
Force two decimal places python
Python