/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 vote152 views1 answer
Get hash python
Python
1 vote126 views1 answer
Use python type hint for multiple return values
Python
0 vote109 views1 answer
%%writefile jupyter notebook
Python
0 vote108 views1 answer
Rotate matrix 90 degrees clockwise in python
Python
6 votes113 views4 answers
Media django
Python
1 vote124 views1 answer
Render_template not showing images
Python
0 vote175 views1 answer
Conda enviroment python version
Python
0 vote112 views1 answer
How to implement nfa in python
Python
0 vote127 views1 answer
Pouqoua sa march pô python ma fonction elle veut pas s'afficher aled
Python
3 votes160 views3 answers
User input dictionary python
Python