如何查看你使用的pytorch是否为GPU版本

1 最简单的方法

>>> import  torch 
 >>> torch.cuda.is_available()
 若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本

2 查看GPU型号的方法

>>> import  torch 
 >>> print(torch.cuda.get_device_name(0))
 返回的GPU型号为你使用的GPU型号

3 直接查看GPU型号的方法

打开终端在终端中输入如下命令：

nvidia-smi

4 查看cuda版本的方法

nvcc -V