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 vote166 views1 answer
Convert unit dynamo revit
Python
0 vote183 views1 answer
Doomsday fuel foobar
Python
1 vote181 views1 answer
Python list .remove() in for loop breaks
Python
0 vote186 views1 answer
Create and activate virtual environment with python 3
Python
0 vote182 views1 answer
Json on desktop python
Python
1 vote167 views1 answer
Pywhatkit
Python
0 vote205 views1 answer
Time complexity for random in python
Python
0 vote173 views1 answer
Modeltranslation
Python
0 vote174 views1 answer
Np arange shape
Python
0 vote169 views1 answer
Setting up a django prostgress project
Python