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

12 votes337 views2 answers
Python input integer only
Python
103 votes309 views4 answers
Separate a string in python
Python
0 vote363 views1 answer
Get admin url of instance django
Python Django
62 votes318 views6 answers
Sort dictionary by values
Python
0 vote290 views1 answer
Print e
Python
35 votes335 views6 answers
How to create an array in python
Python
14 votes271 views2 answers
What does related name do in django
Python
1 vote301 views1 answer
Array and list in python
Python
0 vote300 views1 answer
Write Python programs to print numbers from 1 to 10000 while loops
Python
0 vote326 views1 answer
MEMORY MANAGEMENT SYSTEM IN PYTHON
Python