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

0 vote312 views1 answer
Compare two data frames in assert
Python
1 vote319 views1 answer
Python numpy array size of n
Python
0 vote329 views1 answer
Aiohttp get
Python
0 vote331 views1 answer
Return a new RDD containing only the elements that satisfy a predicate.
Python
0 vote337 views2 answers
Reload class module python
Python
20 votes328 views1 answer
ParserError: Error tokenizing data. C error: Expected 1 fields in line 87, saw 2
Python
0 vote372 views1 answer
Streamlit headings;streamlit text
Python
0 vote371 views2 answers
Generate random password django
Python Django
0 vote318 views1 answer
Doctest python
Python
0 vote378 views1 answer
Flask sqlalchemy case insensitive like
Python