Wie stelle ich fest ob PyTorch meine CUDA GPU nutzt

Question

Wie stelle ich fest ob PyTorch meine GPU nutzt?

Answer 1

Hier ein paar hilfreiche Funktionen

>>> import torch

>>> torch.cuda.is_available()
True

>>> torch.cuda.device_count()
1

>>> torch.cuda.current_device()
0

>>> torch.cuda.device(0)
<torch.cuda.device at 0x7efce0b03be0>

>>> torch.cuda.get_device_name(0)
'GeForce GTX 950M'