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

30 votes263 views5 answers
Python iterate through files in directory
Python
5 votes305 views1 answer
Python convert remove spaces from beginning of string
Python
0 vote251 views1 answer
If else usage python
Python
1 vote252 views1 answer
Python threading vs multiprocessing
Python
68 votes267 views5 answers
Python main method
Python
0 vote251 views1 answer
Python set grid thickness
Python
0 vote257 views3 answers
Python unittest mock
Python
0 vote276 views1 answer
Python DES
Python
0 vote264 views1 answer
Isprime lambda python
Python
0 vote249 views1 answer
Fill na with average pandas
Python