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

2 votes252 views3 answers
How to reboot a python script
Python
9 votes256 views2 answers
Django app
Python Django
0 vote252 views1 answer
Add columns not in place
Python
0 vote300 views1 answer
Python extract all characters from string before a character
Python
0 vote258 views1 answer
Python get name of vlue
Python
2 votes248 views3 answers
Normalize numpy array
Python
0 vote255 views1 answer
How to fix snap "pycharm-community" has "install-snap" change in progress
Python
26 votes234 views6 answers
Python push to list
Python
-2 votes245 views1 answer
Python quiz answer stores
Python
24 votes273 views6 answers
Static files in django
Python