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 vote352 views1 answer
Remove 0 from date in python
Python
2 votes314 views2 answers
Python replace only first instance
Python
0 vote357 views1 answer
Restaurant management system in django
Python Django
8 votes334 views3 answers
Remove the First Character From the String in Python Using the Slicing
Python
0 vote332 views1 answer
Numpy array with 2 times each value
Python
0 vote320 views1 answer
How to put 2 code n 1 line in python
Python
1 vote342 views1 answer
How to write words on any other apps in python
Python
60 votes291 views6 answers
Get current date in python
Python
0 vote355 views1 answer
No module named 'deeppavlov.skills.pattern_matching_skill'
Python
1 vote307 views1 answer
Taking string input from user in python with try except
Python