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

3 votes129 views1 answer
Np append row
Python
0 vote125 views1 answer
Bot that only responds to certain roles discord.py
Python
13 votes142 views1 answer
Selenium webdriver python
Python
13 votes147 views5 answers
How to check if an input is a number in python
Python
0 vote119 views1 answer
Choose value none in pandas
Python
0 vote129 views1 answer
Quality Control in python
Python
1 vote145 views1 answer
NameError: name 'TimeDistributed' is not defined
Python
2 votes161 views1 answer
How to convert gb to mb in python
Python
0 vote136 views6 answers
Gdal user with anaconda
Python
0 vote127 views1 answer
Corresponding angles
Python