How do I check if PyTorch is using the GPU?

by

These functions should help:

>>> 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'

This tells us:

  • CUDA is available and can be used by one device.
  • Device 0 refers to the GPU GeForce GTX 950M, and it is currently chosen by PyTorch.

More Related Contents:

  1. How do I determine the size of an object in Python?
  2. Best way to save a trained model in PyTorch? [closed]
  3. Python memory leaks [closed]
  4. When are objects garbage collected in python?
  5. Does Python have a stack/heap and how is memory managed?
  6. How do I initialize weights in PyTorch?
  7. Why do we need to call zero_grad() in PyTorch?
  8. Python object deleting itself
  9. How does Pytorch’s “Fold” and “Unfold” work?
  10. What does model.eval() do in pytorch?
  11. How to delete multiple pandas (python) dataframes from memory to save RAM?
  12. Python Slice Assignment Memory Usage
  13. Pytorch – Inferring linear layer in_features
  14. Leveraging “Copy-on-Write” to Copy Data to Multiprocessing.Pool() Worker Processes
  15. Pytorch – RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed
  16. Is freeing handled differently for small/large numpy arrays?
  17. Is there a way to delete created variables, functions, etc from the memory of the interpreter?
  18. Best way to join two large datasets in Pandas
  19. Why do two identical lists have a different memory footprint?
  20. How to tell PyTorch to not use the GPU?
  21. How do I visualize a net in Pytorch?
  22. How do I measure the memory usage of an object in python?
  23. Is .data still useful in pytorch?
  24. How to construct a network with two inputs in PyTorch
  25. RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
  26. How can I fix this strange error: “RuntimeError: CUDA error: out of memory”?
  27. Data Augmentation in PyTorch
  28. Google Colaboratory: misleading information about its GPU (only 5% RAM available to some users)
  29. Create a figure that is reference counted
  30. Install PyTorch from requirements.txt

Leave a Comment