Pytorch check if using gpu

Pytorch check gpu

In [1]: import torch

In [2]: torch.cuda.current_device()
Out[2]: 0

In [3]: torch.cuda.device(0)
Out[3]: <torch.cuda.device at 0x7efce0b03be0>

In [4]: torch.cuda.device_count()
Out[4]: 1

In [5]: torch.cuda.get_device_name(0)
Out[5]: 'GeForce GTX 950M'

In [6]: torch.cuda.is_available()
Out[6]: True

Source: stackoverflow.com

Pytorch check if using gpu

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

Pytorch check GPU

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
#或device = torch.device("cuda:0")
device1 = torch.device("cuda:1")  
for batch_idx, (img, label) in enumerate(train_loader):
    img=img.to(device)
    label=label.to(device)

Source: feiguyunai.com

Similar codes

4 votes335 views1 answer
Tkinter text in canvas
Python
0 vote336 views1 answer
Python - How To Pad String With Spaces
Python
2 votes342 views2 answers
Add text to plot python
Python
17 votes340 views4 answers
Python delete directory contents
Python
40 votes350 views6 answers
Pandas sort columns by name
Python
10 votes364 views4 answers
Set method in python
Python
1 vote373 views2 answers
Django url static
Python Django
28 votes360 views2 answers
Drop multiple columns in python
Python
0 vote348 views1 answer
Dependency on app with no migrations:
Python
1 vote332 views1 answer
Python volver al principio
Python