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 vote165 views2 answers
Opencv python convert rgb to hsv
Python
0 vote160 views1 answer
Python render_template
Python
0 vote171 views1 answer
Paschat opposite sanskrit
Python
8 votes227 views3 answers
Python subtract one list from another
Python
0 vote160 views1 answer
Djanog userformcreation message
Python
5 votes162 views3 answers
How to print a message in python
Python
6 votes196 views3 answers
Sort a dictionary by value then key
Python
0 vote165 views1 answer
Python concatenate tuple to string
Python
0 vote188 views1 answer
Save list pickle
Python
-2 votes170 views1 answer
Python check set empty
Python