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 vote283 views1 answer
Python filter list of int and strings
Python
4 votes282 views1 answer
Print specific list item python
Python
0 vote282 views1 answer
Splitting strings in Python without split()
Python
0 vote275 views1 answer
Captions overlap in seaborn plot jupyter
Python
0 vote282 views3 answers
Remove nana from np array
Python
0 vote283 views1 answer
Set up splinter
Python
1 vote356 views1 answer
Robot append to list with for loop
Python
3 votes287 views2 answers
Pandas sum multiple columns groupby
Python
-1 vote285 views2 answers
Print tables in python
Python
0 vote339 views1 answer
Celery beat command for django celery beat
Python Django