How to check whether the pytorch you are using is a GPU version

1 The easiest way

>>> import  torch 
 >>> torch.cuda.is_available()
  If the return is True, We use the GPU Version of torch, if False, Then for CPU edition 

2 see GPU Method of model

>>> import  torch 
 >>> print(torch.cuda.get_device_name(0))
  Back to GPU The model is yours GPU model 

3 Direct view GPU Method of model

Open the terminal and input the following commands in the terminal ：

nvidia-smi

4 see cuda Version method

nvcc -V