python - ¿Cómo comprobar si Pytorch está utilizando la GPU?




3 Answers

Esto va a funcionar :

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'

Esto me dice que la GPU GeForce GTX 950M está siendo utilizada por PyTorch .

Me gustaría saber si pytorch está utilizando mi GPU. Es posible detectar con nvidia-smi si hay alguna actividad de la GPU durante el proceso, pero quiero algo escrito en un script de python .

Hay una manera de hacerlo ?




En el sitio de la oficina y en la página de inicio, verifique la GPU para PyTorch de la siguiente manera:

import torch
torch.cuda.is_available()

Referencia: PyTorch | Comenzar




Cree un tensor en la GPU de la siguiente manera:

$ python
>>> import torch
>>> print(torch.rand(3,3).cuda()) 

No salga, abra otro terminal y verifique si el proceso de python está utilizando la GPU utilizando:

$ nvidia-smi



Related

python memory-management gpu nvidia pytorch