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 vote201 views1 answer
Merge subplot matplotlib
Python
0 vote182 views1 answer
Add zero in front of numbers lower than 10 python
Python
0 vote191 views2 answers
Django create view filter options
Python
0 vote200 views1 answer
MULTIPLE Model Search in Django rest framework
Python
6 votes201 views3 answers
How to remove all zeros from a list in python
Python
1 vote208 views1 answer
Django don't redirect on submission
Python Django
5 votes192 views1 answer
Python leap year solution
Python
24 votes192 views3 answers
Get first x characters of string python
Python
4 votes203 views5 answers
Current time python
Python
0 vote190 views1 answer
Flask Upload file to local s3
Python