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

1 vote132 views1 answer
How to put in code to download discord py
Python
0 vote172 views1 answer
Precision and recall from confusion matrix python
Python
9 votes132 views5 answers
How to make a window in pygame
Python
8 votes109 views1 answer
Python string indexof
Python
1 vote142 views2 answers
Debugar python
Python
27 votes148 views6 answers
Get all file in folder Python
Python
0 vote108 views1 answer
Concat list of strings python
Python
3 votes145 views1 answer
Starting vscode on colab
Python
3 votes140 views3 answers
How to get median mode average of a python list
Python
12 votes136 views6 answers
Check for missing values in pandas
Python