Test cuda pytorch

import torch

torch.cuda.is_available()
>>> True

torch.cuda.current_device()
>>> 0

torch.cuda.device(0)
>>> <torch.cuda.device at 0x7efce0b03be0>

torch.cuda.device_count()
>>> 1

torch.cuda.get_device_name(0)
>>> 'GeForce GTX 950M'

Source: stackoverflow.com

Similar codes

-1 vote351 views1 answer
PyPDF2 Python PDF
Python
0 vote337 views1 answer
Python unzip list
Python
0 vote274 views2 answers
Python del list
Python
0 vote326 views1 answer
Plot row vs column in dataframe python
Python
2 votes381 views1 answer
Django static media
Python Django
2 votes287 views1 answer
Python get all methods of object
Python
-1 vote354 views2 answers
Ai assistant in python
Python
0 vote348 views1 answer
How to change the colour of axes in matplotlin
Python
0 vote404 views1 answer
How to get table schema sql pyodbc
Python
0 vote341 views1 answer
Display full length jupyter
Python