Using GPU with PyTorch

install_and_validate.txt

1- Install PyTorch from the below link. Make sure you select the right CUDA version installed on your device:
https://pytorch.org/

2- After a successful installation. Feel free to test the following commands in your Command Line:
C:\Users\AhmedBr>python
Python 3.6.6 (v3.6.6:4cf1f54eb7, Jun 27 2018, 03:37:03) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.current_device()
0
>>> torch.cuda.device(0)
<torch.cuda.device object at 0x0000028282DBAEF0>
>>> torch.cuda.device_count()
1
>>> torch.cuda.get_device_name(0)
'GeForce 930MX'
>>> torch.cuda.is_available()
True