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 vote128 views1 answer
Passport parsing python
Python
5 votes124 views4 answers
Shebang python
Python
11 votes171 views5 answers
Change default python version mac
Python
0 vote125 views2 answers
Read_json
Python
43 votes139 views2 answers
Write python program to take command line arguments (word count).
Python
0 vote170 views1 answer
Get date only from datetimefiel django
Python Django
0 vote129 views1 answer
Dask dataframe csv tutorial
Python
25 votes153 views3 answers
Check if a value in dataframe is nan
Python
2 votes162 views2 answers
How to convert string to byte without encoding python
Python
0 vote190 views1 answer
Form data postman don't work
Python