1.查看pytorch是否存在以及pytorch的版本

import torch
print(torch.__version__)

 2.查看cuda是否可用

print(torch.cuda.is_available())

3.查看cuda版本

print(torch.version.cuda)

  

4.查看当前工作的GPU 

print(torch.cuda.current_device())

 5.查看gpu的数目

print(torch.cuda.device_count())

6.设置工作GPU

print(torch.cuda.current_device())
torch.cuda.set_device(1)
print(torch.cuda.current_device())

也有的方法说下述方式也可以实现切换GPU但本人亲测无效

os.environ["CUDA_VISIBLE_DEVICES"] = '0'

 

7.查看gpu名字，设备索引默认从0开始：

print(torch.cuda.get_device_name(0))
print(torch.cuda.get_device_name(1))