/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

0 vote256 views1 answer
Django require_POST body
Python Django
1 vote219 views2 answers
Where
Python
6 votes260 views6 answers
Install pip python 3.9
Python
4 votes219 views3 answers
What is a character in python
Python
0 vote198 views1 answer
Display vowels in a string using for loop
Python
0 vote259 views1 answer
Fetch json array from mysql django
Python Django
11 votes258 views3 answers
Shuffle dataframe python
Python
0 vote195 views2 answers
Numpy index of first true value
Python
1 vote288 views1 answer
Pyton count number of character in a word
Python
1 vote183 views1 answer
How to find the neighbors of an element in matrix python
Python