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

4 votes289 views8 answers
Play solitaire
Python
0 vote212 views1 answer
Nornir config file
Python
0 vote238 views1 answer
How to check if a cell is empty in openpyxl
Python
0 vote247 views1 answer
Django listview
Python Django
2 votes228 views1 answer
MAPE python
Python
6 votes232 views5 answers
Python open folder in explorer
Python
3 votes228 views2 answers
Matplotlib log scale y axis base
Python
0 vote216 views1 answer
Matplotlib transparent line
Python
0 vote215 views1 answer
Subtract from dataframe
Python
1 vote249 views1 answer
ValueError: logits and labels must have the same shape ((None, 1) vs (None, 2))
Python