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

4 votes217 views1 answer
Tkinter text in canvas
Python
27 votes245 views4 answers
Random letter generator python
Python
0 vote238 views1 answer
"DjangoJSONEncoder" is not defined
Python Django
7 votes212 views2 answers
Select column in pandas dataframe
Python
0 vote211 views1 answer
Change set item python
Python
0 vote249 views1 answer
Selenium press tab python
Python
1 vote207 views1 answer
Jupyter notebook no password or token
Python
0 vote197 views1 answer
Pandas currency to numbe
Python
1 vote217 views1 answer
USB: usb_device_handle_win.cc:1049 Failed to read descriptor from node connection: A device attached to the system is not functioning. (0x1F)
Python
2 votes199 views1 answer
Copy website python
Python